擅长:python、mysql、java
<p>你只需要在主循环中的粒子数上增加一个循环。在</p>
<p>为了便于阅读,我还在下面的代码片段中创建了一个函数来进行速度计算:</p>
<pre><code>import math
all_velocities = [[[1,2,3], [2,3,4]], [[3,4,5], [4,5,6], [7,8,9]]]
all_speeds = []
def calculate_speed(v):
return math.sqrt(v[0]**2 + v[1]**2 + v[2]**2)
for i in range(len(all_velocities)):
all_speeds.append([])
for j in range(len(all_velocities[i])):
velocity = all_velocities[i][j]
all_speeds[i].append(calculate_speed(velocity))
print all_speeds
</code></pre>