python-numpy中axis的理解

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章