numpy中axis理解
阿新 • • 發佈:2019-01-07
numpy庫中有些函式有個引數axis,像ndarray.max()等,ndarray 其實就是一個多維陣列,比python 內建的list多了一些對多維陣列的操縱方法。如果一個多維陣列是2*3*2,他就是3維的,引數axis=0指的就是第一維,即2那一個數軸,axis=1,即3的那個數軸,axis=2,即最後2的那個數軸。
程式段
import numpy as np
np.random.seed(123)
x=np.random.randint(0,5,[2,3,2])
print x
print x.max(axis=0)
輸出
[[[2 4] [2 1] [3 2]] [[3 1] [1 0] [1 1]]] [[3 4] [2 1] [3 2]]
分析:
指定axis=0,即在第一維的數軸求取最大值,2*3*2,去掉第一個維度,結果應該是3*2。將上面x的上面和下面塊比較2->3,4->1,2->1,1->0,3->1,2->1就可以啦。
如果指定axis=1,則輸出為
[[[2 4]
[2 1]
[3 2]]
[[3 1]
[1 0]
[1 1]]]
[[3 4]
[3 1]]
將2-2-3,4-1-2,3-1-1,1-0-1進行比較就可以啦,同理axis=2