numpy sum along axis

Question

Is there a numpy function to sum an array along (not over) a given axis? By along an axis, I mean something equivalent to:

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

to sum along axis i.

For example, a case where numpy.sum will not work directly:

>>> 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]])

Does this answer your question? How does NumPy Sum (with axis) work? — Karl Knechtel
– Karl Knechtel, Commented Dec 2, 2023 at 7:47

jeroko · Accepted Answer · 2015-10-27 00:18:15Z

7

You can just pass a tuple with the axes that you want to sum over, and leave out the one that you want to 'sum along':

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

answered Oct 27, 2015 at 0:18

jeroko

4486 silver badges9 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

JoeZuntz · Accepted Answer · 2014-08-08 12:57:57Z

4

As of numpy 1.7.1 there is an easier answer here - you can pass a tuple to the "axis" argument of the sum method to sum over multiple axes. So to sum over all except the given one:

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

answered Aug 8, 2014 at 12:57

JoeZuntz

1,18910 silver badges17 bronze badges

Comments

Owen · Accepted Answer · 2011-08-21 22:24:55Z

3

Call sum twice?

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

Of course, this would be a little awkward to generalize because axes "disappear". Do you need it to be general?

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

answered Aug 21, 2011 at 22:24

Owen

39.5k14 gold badges98 silver badges131 bronze badges

Comments

user545424 · Accepted Answer · 2012-02-08 18:36:25Z

2

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)

answered Feb 8, 2012 at 18:36

user545424

16.3k11 gold badges61 silver badges72 bronze badges

Comments

fshdky · Accepted Answer · 2012-02-12 05:46:35Z

2

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

edited Feb 12, 2012 at 5:46

answered Feb 12, 2012 at 5:36

fshdky

212 bronze badges

Collectives™ on Stack Overflow

numpy sum along axis

5 Answers 5

Comments

Comments

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

5 Answers 5

Comments

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related