2013-10-09 20 views

cevap

4

Sen yayına kullanabilirsiniz:

c = sum(a[...,np.newaxis]*b[np.newaxis,...],axis=1) # == np.dot(a,b) 

c = sum(a[...,np.newaxis]==b[np.newaxis,...],axis=1) 

O dizi nasıl genişlediğini sadece şunu açıklığa kavuşturalım b içinde newaxis dahil. Dizilere boyut eklemenin başka yolları vardır (yeniden şekillendirmek, tekrarlamak, vb.), Ancak etki aynıdır. Öğeyi çarparak (veya ==) öğeye yapmak için aynı şekle a ve b genişletin ve sonra doğru eksende toplayın.

+0

+1 nice one, teşekkürler –

+0

Teşekkürler. Bu çok temiz. –