文章目录
版权声明:本文为博主原创文章,转载请注明原文出处!
写作时间:2019-04-16 14:56:53
浅谈NumPy中的维度Axis
NumPy中的维度是一个很重要的概念,很多函数的参数都需要给定维度Axis,如何直观的理解维度呢?我们首先以二维数组为例进行说明,然后推广到多维数组。
(有人将ndim
属性叫维度,将axis
叫轴,我还是习惯将axis
称之为维度,axis=0
称为第一个维度)
二维数组的列子
下面是一个二维数组的列子:
In [1]: import numpy as np
In [2]: x = np.random.randint(0, 9, (2, 3))
In [3]: x
Out[3]:
array([[0, 8, 6],
[1, 2, 1]])
In [4]: x.ndim
Out[4]: 2
In [5]: x.shape
Out[5]: (2, 3)
In [6]: x[0]
Out[6]: array([0, 8, 6])
In [7]: x[:, 0]
Out[7]: array([0, 1])
In [8]: x.sum(axis=0)
Out[8]: array([ 1, 10, 7])
In [9]: x.sum(axis=1)
Out[9]: array([14, 4])
In [10]: x[0] + x[1]
Out[10]: array([ 1, 10, 7])
In [11]: x[:, 0] + x[:, 1] + x[:, 2