import numpy as np
if __name__ == '__main__':
a = np.arange(9).reshape((3, 3))
print(a)
#每列最小
print(np.min(a, axis=0))
#每列最大
print(np.max(a, axis=0))
#每行最小
print(np.min(a, axis=1))
#每行最大
print(np.max(a, axis=1))
结果:
[[0 1 2]
[3 4 5]
[6 7 8]]
[0 1 2]
[6 7 8]
[0 3 6]
[2 5 8]