今天在看代碼時,發現一個小地方,就是求每行每列的最大最小元素時,感覺對軸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---