import numpy as np
c=[10,2000] # only two index for example
horizontal=[]
vertical=[]
for i in range(0,c[0]):
for j in range(0, c[1]):
horizontal.append(j)
vertical.append(i)
print horizontal
当我处理数组和循环时,需要花费太多的时间。据我了解,由于未定义的数据类型("i“和"j"),这是在浪费时间。我希望使用NumPy定义数据类型,并加快数组循环。
对于这些“数组for循环”,我如何使用NumPy,或者有没有其他解决方案来加速这些循环?
发布于 2017-06-12 04:22:48
使用NumPy最简单的方法是np.mgrid
,它创建了一个网格:
vertical, horizontal = np.mgrid[0:c[0], 0:c[1]]
vertical = vertical.ravel() # make it 1D
horizontal = horizontal.ravel() # make it 1D
https://stackoverflow.com/questions/44499301
复制