1. 程式人生 > >numpy中axis理解

numpy中axis理解

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