How to stack summing vectors to numpy 3d array?-CodePudding

I have a 3d numpy array that looks like so:

>>> g
array([[[ 1.,  1.,  1.,  1.,  1.],
        [ 0.,  0.,  0.,  0.,  0.],
        [ 1.,  2.,  3.,  4.,  6.]],

       [[ 0.,  0.,  0.,  0.,  0.],
        [11., 22., 33., 44., 66.],
        [ 0.,  0.,  0.,  0.,  0.]]])

I know I can calculate a sum along the first axis with gs = g.sum(axis=1) that will result in this array:

>>> gs
array([[ 2.,  3.,  4.,  5.,  7.],
       [11., 22., 33., 44., 66.]])

How do I stack this summing array to the original one as the forth vector in each of the two inside groups? The expected result would be:

>>> g
array([[[ 1.,  1.,  1.,  1.,  1.],
        [ 0.,  0.,  0.,  0.,  0.],
        [ 1.,  2.,  3.,  4.,  6.],
        [ 2.,  3.,  4.,  5.,  7.]],

       [[ 0.,  0.,  0.,  0.,  0.],
        [11., 22., 33., 44., 66.],
        [ 0.,  0.,  0.,  0.,  0.],
        [ 11., 22., 33., 44., 66.]]])

And I have the same question about the summing array along the 0th dimension which is calculated with gss = g.sum(axis=0) and looks like so:

>>> gss
array([[ 1.,  1.,  1.,  1.,  1.],
       [11., 22., 33., 44., 66.],
       [ 1.,  2.,  3.,  4.,  6.]])

How do I stack it to the original array to get the result shown below?

>>> g
array([[[ 1.,  1.,  1.,  1.,  1.],
        [ 0.,  0.,  0.,  0.,  0.],
        [ 1.,  2.,  3.,  4.,  6.]],

       [[ 0.,  0.,  0.,  0.,  0.],
        [11., 22., 33., 44., 66.],
        [ 0.,  0.,  0.,  0.,  0.]],

       [[ 1.,  1.,  1.,  1.,  1.],
        [11., 22., 33., 44., 66.],
        [ 1.,  2.,  3.,  4.,  6.]]])

CodePudding user response：

np.concatenate([g, g.sum(axis=1, keepdims=True)], axis=1)

Equivalent for axis=0

CodePudding user response：

I think this help you with your question:

import numpy as np

g = np.array([[[ 1.,  1.,  1.,  1.,  1.],
        [ 0.,  0.,  0.,  0.,  0.],
        [ 1.,  2.,  3.,  4.,  6.]],

        [[ 0.,  0.,  0.,  0.,  0.],
        [11., 22., 33., 44., 66.],
        [ 0.,  0.,  0.,  0.,  0.]]])

gs = g.sum(axis=1)

g_stacked = np.concatenate((g, gs[:, np.newaxis, :]), axis=1)

print(g_stacked)

gss = np.array([[ 1.,  1.,  1.,  1.,  1.],
       [11., 22., 33., 44., 66.],
       [ 1.,  2.,  3.,  4.,  6.]])

gss = g.sum(axis=0)

g_stacked = np.concatenate((g, gss[np.newaxis, :, :]), axis=0)

print(g_stacked)