Numpy: Làm thế nào để nhân hai phần tử, hình dạng (n, 1) và (n,)?

Question

Phép nhân nhân hai vectơ không phải là vấn đề nếu cả hai đều có cùng hình dạng, nói cả hai (n, 1) hoặc cả hai (n,). Nếu một vectơ có hình dạng (n, 1) và cái kia (n,), mặc dù, * -operator trả về một cái gì đó vui nhộn.

a = np.ones((3,1)) 
b = np.ones((3,)) 
print a * b 

Ma trận nxn kết quả chứa A_ {i, j} = a_i * b_j.

Làm cách nào tôi có thể thực hiện phép nhân số nguyên tố cho số a và b sau đó?

Answer 1

Slice các vectơ trong một cách mà làm cho trận đấu hình dạng của chúng:

a[:, 0] * b 

hoặc

a * b[:, None] 

Answer 2

Thêm một trục thứ hai để b đó a và b có kích thước tương tự:

>>> a * b[:,np.newaxis] 
array([[ 1.], 
     [ 1.], 
     [ 1.]]) 

Cách khác, chuyển đổi a để phát sóng hoạt động:

>>> a.T * b 
array([[ 1., 1., 1.]]) 

(Bạn có thể muốn transpose kết quả.)