Numpy：shapewise（n、1）と（n、）の2つのベクトルを要素に掛ける方法は？

Question

2つのベクトルの要素ごとの乗算は、両方が同じ形状、例えば（n、1）または両方（n、）の両方であれば問題ありません。一方のベクトルがshape（n、1）で他方が（n、）の場合、* -operatorは面白いものを返します。

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

結果のnxn行列には、A_ {i、j} = a_i * b_jが含まれます。

aとbの要素ごとの乗算はどうすればできますか？それらの形状の一致を行うように

Answer 1

スライスベクター：

a[:, 0] * b 

又は

a * b[:, None] 

Answer 2

そのaにbに第2の軸を追加しbは、同じ寸法を有する：

を

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

また、転位aように放送作品：

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

（おそらく結果を移調したいと思います。）