[问题] numpy新手问题 (np相乘与shape变换)

楼主: odanaga (PixiyON)   2018-02-19 19:55:29
新年快乐
小弟python新手想问个numpy问题
今天我有一个shape = (40,3)的A
还有一个shape = (40,1)的B
利用np.multiply可以乘出一个shape = (40,3)的矩阵
若我今天有一个C.shape = (260,40,1)
我要怎样才能(不用for循环)乘出一个(260,40,3)的矩阵呢?
谢谢
作者: iphone2003 (307)   2018-02-19 20:24:00
一样直接np.multiply(A, C)就可了吧 broadcasting不然也可以用A[np.newaxis, :, :]先垫一个维度再做multiply,这样就和第一个例子差不多了c.shape=(260, 40)时不行是因为broadcasting的规则是把大小是1的那个维度拉到跟另外一个阵列一样,那个时候A和C完全没有大小是1的维度所以才无法成功
作者: frish8932   2018-03-02 16:06:00
作者: ar54971   2018-03-06 03:40:00

Links booklink

Contact Us: admin [ a t ] ucptt.com