numpy axis

import numpy as np

# 随机产生240-1的数字
a = np.random.randint(0,9, (3,4))
print(a)

# data =[[a00, a01],[a10,a11]],
# 所以axis=0时,沿着第0个下标变化的方向进行操作,
# 也就是a00->a10, a01->a11,
array_sum = np.sum(a, axis=1)
array_min = np.min(a, axis=0)
array_max = np.max(a, axis=1)

print(array_sum)
print(array_min)
print(array_max)

结果:
[[0 3 6 8]
[3 3 1 1]
[8 0 3 5]]
[17 8 16]
[0 0 1 1]
[8 3 8]

猜你喜欢

转载自blog.csdn.net/code_fighter/article/details/80371912