擅长:python、mysql、java
<pre><code>import numpy as np
v1 = [[[1,1,1],
[2,2,2],
[3,3,3],
[4,4,4]],
[[1,1,1],
[2,2,2],
[3,3,3],
[4,4,4]]]
a = np.array(v1)
</code></pre>
<p><code>a.shape</code>是(t,n,d)。你需要三维速度平方和的平方根:</p>
^{pr2}$
<p><code>b.shape</code>是(t,n)。在我的数据集中似乎没有粒子在加速。在</p>
<p>或者简单地说:</p>
<pre><code>>>> np.linalg.norm(a, axis = 2)
array([[ 1.73205081, 3.46410162, 5.19615242, 6.92820323],
[ 1.73205081, 3.46410162, 5.19615242, 6.92820323]])
>>>
</code></pre>