numpy二维轴和三维轴(axis)

代码示例:

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

猜你喜欢

转载自blog.csdn.net/caoxinjian423/article/details/112075340