代码示例:
import numpy as np
#二维数组
my_np1 = np.arange(18).reshape(3,6)
print(my_np1)
'''
打印:
[[ 0 1 2 3 4 5]
[ 6 7 8 9 10 11]
[12 13 14 15 16 17]]
'''
print(np.sum(my_np1)) #打印153,所有元素的和
print(np.sum(my_np1,axis=0)) #打印[18 21 24 27 30 33],即每列的和组成的一维数组
print(np.sum(my_np1,axis=1)) #打印[15 51 87],即每行的和组成的一维数组
#三维数组
# # 三维
my_np2 = np.arange(27).reshape((3,3,3))
print(my_np2)
'''
[[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]]
[[ 9 10 11]
[12 13 14]
[15 16 17]]
[[18 19 20]
[21 22 23]
[24 25 26]]]
'''
print(np.sum(my_np2, axis=0))
'''
即三个二维子数组相同位置的和组成的二维数组
打印:
[[27 30 33]
[36 39 42]
[45 48 51]]
'''
print(np.sum(my_np2, axis=1))
'''
即每个二维数组的每列和组成结果二维数组的行,比如0 3 6之和组成结果二维数组的第一个元素
打印:
[[ 9 12 15]
[36 39 42]
[63 66 69]]
'''
print(np.sum(my_np2, axis=2))
'''
即每个二维数组的每行和组成结果二维数组的行,比如0 1 2之和组成结果二维数组的第一个元素
打印:
[[ 3 12 21]
[30 39 48]
[57 66 75]]
'''