今天在看代码时,发现一个小地方,就是求每行每列的最大最小元素时,感觉对轴0和1的设置有点转不过弯来,于是写了一小段代码帮助理解。
import numpy as np
a=np.array([[1,5,3],[4,2,6],[-5,4,9],[7,0,-9],[10,-2,6]])
print("初始矩阵:")
print(a)
print("求矩阵中最小元素:")
print(np.min(a))
print("求每行最小元素:")
print(np.min(a,1))
print("求每列最小元素:")
print(np.min(a,0))
关于轴的理解,一定不要和下标给掺了,这里我简单画了一个小的示意图:
---end---