如何向量化以删除Python中的for循环？

如何解决如何向量化以删除Python中的for循环？

我想通过矢量化以下Python代码中的for循环。我在整个互联网上搜索了堆栈溢出问题，以找到无济于事的解决方案。

for v in range(height):
  for u in range(width):
    start[v,u,0] = -0.5 + u / (width-1)
    start[v,1] = (-0.5 + v / (height-1)) * height / width
    start[v,2] = 0

我认为我们应该使用NumPy。也许要使用numpy.linspace（）？所以，你怎么看？

解决方法

以下内容：

start[v,u,0] = -0.5 + u / (width-1)
start[v,1] = (-0.5 + v / (height-1)) * height / width
start[v,2] = 0

如果v和u是花哨的索引，则可以工作：

start = np.random.randint(99,size=(5,7,3)).astype(float)
height,width = start.shape[:2]
v,u = np.indices((height,width))

样本运行（数值四舍五入）：

>>> v
[[0 0 0 0 0 0 0]
 [1 1 1 1 1 1 1]
 [2 2 2 2 2 2 2]
 [3 3 3 3 3 3 3]
 [4 4 4 4 4 4 4]]
>>> u
[[0 1 2 3 4 5 6]
 [0 1 2 3 4 5 6]
 [0 1 2 3 4 5 6]
 [0 1 2 3 4 5 6]
 [0 1 2 3 4 5 6]]
>>> start[v,0]
[[-0.5  -0.33 -0.17  0.    0.17  0.33  0.5 ]
 [-0.5  -0.33 -0.17  0.    0.17  0.33  0.5 ]
 [-0.5  -0.33 -0.17  0.    0.17  0.33  0.5 ]
 [-0.5  -0.33 -0.17  0.    0.17  0.33  0.5 ]
 [-0.5  -0.33 -0.17  0.    0.17  0.33  0.5 ]]
>>> start[v,1]
[[-0.36 -0.36 -0.36 -0.36 -0.36 -0.36 -0.36]
 [-0.18 -0.18 -0.18 -0.18 -0.18 -0.18 -0.18]
 [ 0.    0.    0.    0.    0.    0.    0.  ]
 [ 0.18  0.18  0.18  0.18  0.18  0.18  0.18]
 [ 0.36  0.36  0.36  0.36  0.36  0.36  0.36]]

快速预览以更好地理解：

,

在没有更多信息的情况下很难回答您的问题（要创建数组吗？它已经创建好了吗？为什么您的v，u索引“向后”？...

但是，要回答您的问题，是的-有很多方法可以使用numpy来完成您想做的事情。

下面看一看，然后阅读一两个小时的一些numpy教程，等等：

import numpy as np

rows = 10
cols = 5

values = np.zeros((rows,cols,3))

values[:,:,0] = np.fromfunction(
    lambda r,c: -0.5 + c / (cols - 1),(rows,cols))

values[:,1] = np.fromfunction(
    lambda r,c: (-0.5 + r / (rows - 1) * rows / cols),cols))

请注意，使用numpy可以执行多种操作。

以上只是其中之一，并且以更扩展，更易于遵循的方式编写。您可能可以在一行中执行相同的操作，但是如果您不熟悉numpy，将很难理解。