使用 np.einsum，就像这样 -

Use np.einsum, like so -

np.einsum('ij,ij->j',arr,arr)

样品运行 -

In [243]: np.einsum('ij,ij->j',arr,arr)
Out[243]: array([    101,   10004, 1000009])

或者用 np.sum -

In [244]: (arr**2).sum(0)
Out[244]: array([    101,   10004, 1000009])

或者使用 numexpr 模块 -

Or with numexpr module -

In [248]: import numexpr as ne

In [249]: ne.evaluate('sum(arr**2,0)')
Out[249]: array([    101,   10004, 1000009])