如何在python中简化for循环

Question

我有一段Python代码填充for循环中的2d矩阵

img=zeros((len(bins_x),len(bins_y)))

for i in arange(0,len(ix)):
   img[ix[i]][iy[i]]=dummy[i]

是否可以对最后两行代码使用矢量运算？还有什么东西可以加快计算速度吗？

Answer 1

如果是ix，我是索引序列：

img[ix, iy] = dummy

Answer 2

使用numpy可能很有用。特别是，reshape方法可能很有用。这是一个例子（改编自第二个链接）：

>>> import numpy as np
>>> a = np.array([1,2,3,4,5,6])
>>> np.reshape(a, (3,2))
array([[1, 2],
       [3, 4],
       [5, 6]])