import numpy as np a = np.array( [ [1, 2, 3 ], [4, 5, 6] ] ) print( "a =", a, sep="\n" ) print( "a.sum() =", a.sum() ) print( "a.sum(axis=0) =", a.sum(axis=0) ) b= a.sum(axis=1) print( "a.sum(axis=1) =", b ) print( "a.sum(axis=1).shape =", b.shape )
a = [[1 2 3] [4 5 6]] a.sum() = 21 a.sum(axis=0) = [5 7 9] a.sum(axis=1) = [ 6 15] a.sum(axis=1).shape = (2,)