像這樣:
import numpy as np
a = np.zeros((3,3))
b = np.ones((3,1))
a[:,2] = b
預期的:
a =
0,0,1
0,0,1
0,0,1
實際上:
ValueError: non-broadcastable output operand with shape (3,) doesn't match the broadcast shape (3,3)
我該怎么辦?
uj5u.com熱心網友回復:
需要指定列的范圍
例如a[:,0:1]對于第 0a[:,1:2]列、第 1a[:,2:]列和第 2 列。
import numpy as np
a = np.zeros((3,3))
b = np.ones((3,1))
a[:,2:] = b
輸出:
array([[0., 0., 1.], [0., 0., 1.], [0., 0., 1.]])
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/383419.html
