擅长:python、mysql、java
<pre><code>files = np.load("particle.npz")
x_s = [files[key] for key in files.keys()]
</code></pre>
<p>在Python中,创建数组列表而不是单独命名的数组是首选方法</p>
<pre><code>pox_s = [x[:,0] for x in x_s]
</code></pre>
<p>看起来数组的大小都一样。因此,我们可以将列表转换为数组:</p>
<pre><code>pox_s = np.array(pox_s)
</code></pre>
<p>甚至</p>
<pre><code>x_s = np.array(x_s) # (6, 40002, ?)
pox_s = x_s[:,:,0] # (6, 40002)
sq_diffs = (pox_s - pox_s[:,[0]])**2 # (6, 40002)-(6,1)
</code></pre>
<p>如果没有一个具体的小例子,我就无法测试这段代码。我想我的形状是对的</p>