numpy 沿着轴的数值和

rqqzpn5f 于 12个月前发布在其他

关注(0)|答案(5)|浏览(136)

有没有一个numpy函数可以对一个数组 * 沿着 *（而不是超过）一个给定的轴求和？通过沿着一个轴，我的意思是等价于：

[x.sum() for x in arr.swapaxes(0,i)].

字符串
沿沿着 * 轴i求和 *。
例如，numpy.sum不能直接工作的情况：

>>> a = np.arange(12).reshape((3,2,2))
>>> a
array([[[ 0,  1],
        [ 2,  3]],

       [[ 4,  5],
        [ 6,  7]],

       [[ 8,  9],
        [10, 11]]])
>>> [x.sum() for x in a] # sum along axis 0
[6, 22, 38]
>>> a.sum(axis=0)
array([[12, 15],
       [18, 21]])
>>> a.sum(axis=1)
array([[ 2,  4],
       [10, 12],
       [18, 20]])
>>> a.sum(axis=2)
array([[ 1,  5],
       [ 9, 13],
       [17, 21]])

型

numpy

来源：https://stackoverflow.com/questions/7140560/numpy-sum-along-axis

5条答案

按热度按时间

aelbi1ox1#

你可以只传递一个包含你想要求和的轴的元组，而忽略你想要“求和”的轴：

>> a.sum(axis=(1,2))
array([ 6, 22, 38])

字符串

赞(0）回复(0）举报 12个月前

xkftehaa2#

从numpy 1.7.1开始，这里有一个更简单的答案-你可以将一个元组传递给sum方法的“axis”参数，以在多个轴上求和。因此，除了给定的轴之外，对所有轴求和：

x.sum(tuple(j for j in xrange(x.ndim) if j!=i))

字符串

赞(0）回复(0）举报 12个月前

nafvub8i3#

打两次电话给萨姆？

In [1]: a.sum(axis=1).sum(axis=1)
Out[1]: array([ 6, 22, 38])

字符串
当然，这样概括起来会有点尴尬，因为坐标轴“消失”了。你需要它是通用的吗？

def sum_along(a, axis=0):
    js = [axis] + [i for i in range(len(a.shape)) if i != axis]
    a = a.transpose(js)

    while len(a.shape) > 1: a = a.sum(axis=1)

    return a

型

赞(0）回复(0）举报 12个月前

c9qzyr3d4#

def sum_along_axis(a, axis=None):
    """Equivalent to [x.sum() for x in a.swapaxes(0,axis)]"""
    if axis is None:
        return a.sum()
    return np.fromiter((x.sum() for x in a.swapaxes(0,axis)), dtype=a.dtype)

字符串

赞(0）回复(0）举报 12个月前

yshpjwxd5#

np.apply_over_axes(sum, a, [1,2]).ravel()

字符串

赞(0）回复(0）举报 12个月前

我来回答

numpy 沿着轴的数值和

5条答案

相关问题

热门标签

最新问答