擅长:python、mysql、java
<p>再添加一个循环</p>
<pre><code># Takes all velocities and converts them to speeds
all_speeds=[]
# Looping over time
for i in range(len(all_velocities)):
#adding and empty list to collect all the speeds
all_speeds.append([])
#looping over number of particles
for j in range(len(all_velocities[i])):
all_speeds[-1].append([math.sqrt(all_velocities[i][j][0]**2\
+all_velocities[i][j][1]**2\
+all_velocities[i][j][2]**2)])
</code></pre>
<p>希望这有帮助。在</p>