假設矩陣是2d numpy (MxN)陣列，向量是1d陣列(1xN)--兩者有相同的N行。我需要在矩陣的每一列中加入與向量中相同的行元素的值：
[[1 2 3] 。
[4 5 6 ]]
[0.1 0.2]
結果。
[[1.1 2.1 3.1] 。
[4.2 5.2 6.2 ]]
uj5u.com熱心網友回復：
你可以使用numpy.reshape(-1,1)并得到你想要的東西：
l1 = np. array([[1, 2, 3], [4, 5 , 6]])
l2 = np.array([0.1, 0.2]).reshape（-1,1)
l1 l2
輸出：
array([[1.1, 2.1, 3.1],
[4.2, 5.2, 6.2]])
uj5u.com熱心網友回復：
利用numpy廣播的優勢，你可以做a b[:, None]：
a = np.range(1,7).reshape((2, 3)
b = np.array([0.1, 0.2] )
a b[:, None]
array([[1.1, 2.1, 3.1],
[4.2, 5.2, 6.2]])
或者：
a b[:, np.newaxis] 。
array([[1.1, 2.1, 3.1],
[4.2, 5.2, 6.2]])
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/309487.html
標籤：