将 Numpy 应用于“for 循环”

Question

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

当我使用数组和 for 循环时，会花费太多时间。据我了解，由于未定义的数据类型（“i”和“j”），这是在浪费时间。我想使用 NumPy 定义数据类型并加快数组循环。

如何将 NumPy 用于这些“循环数组”，或者是否有任何其他解决方案可以加快这些循环？

Answer 1

使用 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