j=np.arange(20,dtype=np.int)
site=np.ones((20,200),dtype=np.int)
sumkma=np.ones((100,20))
[sumkma[site[x],x] for x in range(20)]
这很有效,但我不想用于循环。当我试着
sumkma[site[j],j]
我得到这个错误:
IndexError: shape mismatch: indexing arrays could not be broadcast together with shapes (20,200) (20,)
如何修正错误?
当使用其他整数类型的多维数组访问numpy多维数组时,用于索引的数组需要具有相同的形状。
如果可能的话,Numpy会很高兴地广播,但是为了实现这一点,数组需要具有相同的维数,例如:
np.newaxis
导致j[:,np.newaxis]
是二维的(形状现在是(20,1)
,而j
的形状是一维的(20,)
)。它现在有一个可以广播到site[j]
形状的形状:索引数组的相同维度允许numpy广播它们具有相同的形状
(20,200)
。https://docs.scipy.org/doc/numpy-1.13.0/user/basics.indexing.html#indexing-multi-dimensional-arrays
相关问题 更多 >
编程相关推荐