擅长:python、mysql、java
<p>使用<a href="https://numpy.org/doc/stable/reference/generated/numpy.atleast_2d.html" rel="nofollow noreferrer">^{<cd1>}</a>-</p>
<pre><code>np.atleast_2d(ar).mean(axis=0)
</code></pre>
<p>对于2D,<code>np.atleast_2d</code>不会改变任何东西。对于<code>1D</code>,让我们看一个示例案例-</p>
<pre><code>In [125]: a1D = np.arange(4).astype(float)
In [126]: a1D
Out[126]: array([0., 1., 2., 3.])
In [127]: np.atleast_2d(a1D).mean(axis=0)
Out[127]: array([0., 1., 2., 3.])
</code></pre>
<p>另一个与重塑-</p>
<pre><code>ar.reshape(-1,ar.shape[-1]).mean(0)
</code></pre>