python之numpy之axis=1和axis=0

对二维数组而言，一般来说axis=1表示对行进行操作，axis=0表示对列进行操作。
以sort¹、mean²为例进行描述

sort

>>> import numpy as np
>>> a = np.array([[1,4],[3,1]])
>>> np.sort(a)                # sort along the last axis
array([[1, 4],
       [1, 3]])
>>> np.sort(a, axis=None)     # sort the flattened array
array([1, 1, 3, 4])
>>> np.sort(a, axis=0)        # sort along the first axis
array([[1, 1],
       [3, 4]])

mean

>>> import numpy as np
>>> a = np.array([[1, 2], [3, 4]])
>>> np.mean(a)
2.5
>>> np.mean(a, axis=0)
array([2., 3.])
>>> np.mean(a, axis=1)
array([1.5, 3.5])

---

1. SciPy官方：numpy.sort ↩︎

2. SciPy官方：numpy.mean ↩︎