5

到目前为止，我从来没有担心过这个问题，但现在我需要使用大量需要由 PyOpenGL 缓冲的顶点，而且似乎 python 迭代是瓶颈。这是情况。我有一个 3D 点数组vertices，在每一步我都必须为每个顶点计算一个 4D 颜色数组。到目前为止，我的方法是：

upper_border = len(self.vertices) / 3
#Only generate at first step, otherwise use old one and replace values
if self.color_array is None:
     self.color_array = numpy.empty(4 * upper_border)  

for i in range(upper_border):
     #Obtain a color between a start->end color
     diff_activity = (activity[i] - self.min) / abs_diff  
     clr_idx = i * 4
     self.color_array[clr_idx] = start_colors[0] + diff_activity * end_colors[0]
     self.color_array[clr_idx + 1] = start_colors[1] + diff_activity * end_colors[1]
     self.color_array[clr_idx + 2] = start_colors[2] + diff_activity * end_colors[2]
     self.color_array[clr_idx + 3] = 1

现在我认为我不能做任何其他事情来消除循环的每个步骤中的操作，但我猜必须有一种更优化的性能方式来执行该循环。我之所以这么说是因为例如在 javascript 中，相同的演算会产生 9FPS，而在 Python 中我只能得到 2-3 FPS。

问候， 博格丹

4

2 回答 2

13

为了使这段代码更快，您需要对其进行“矢量化”：使用 NumPy 的广播规则将所有显式 Python 循环替换为隐式循环。我可以尝试给出循环的矢量化版本：

if self.color_array is None:
     self.color_array = numpy.empty((len(activity), 4))
diff_activity = (activity - self.min) / abs_diff
self.color_array[:, :3] = (start_colors + 
                           diff_activity[:, numpy.newaxis] + 
                           end_colors)
self.color_array[:, 3] = 1

请注意，我不得不做很多猜测，因为我不确定你所有的变量是什么以及代码应该做什么，所以我不能保证这段代码会运行。我变成color_array了一个二维数组，因为这样看起来更合适。这可能需要更改代码的其他部分（或者您需要再次展平数组）。

我假设self.min和abs_diff是标量，所有其他名称都引用以下形状的 NumPy 数组：

activity.shape == (len(vertices) // 3,)
start_colors.shape == (3,)
end_colors.shape == (3,)

它也看起来好像vertices是一个一维数组，应该是一个二维数组。

于 2011-12-12T14:40:13.013 回答
6
  1. 首先：使用cProfile分析您的代码
  2. 您应该使用xrange而不是 range
  3. 您应该避免self.color_array在每个循环中调用 4 次，尝试在循环之前创建一个局部变量，并将其用于循环：local_array = self.color_array
  4. 尝试预先计算start_colors[N]and end_colors[N]：start_color_0 = start_colors[0]

  5. 尝试使用list.extend()减少循环中的行数：

    local_array.extend([
   start_colors_0 + diff_activity * end_colors_0,
   start_colors_1 + diff_activity * end_colors_1,
   start_colors_2 + diff_activity * end_colors_2,
   1
])
于 2011-12-12T14:39:09.630 回答