原创

NumPy中的维度Axis

版权声明:本文为博主原创文章,转载请注明原文出处!

写作时间: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]
Out[11]: array([14,  4])

看上面这个例子,x是一个2行3列的数组,所以x是一个二维数组。

从第6和第7个输入输出,我们可以肯定地说"对于二维数组,第一维指的是行,第二维指的是列"。

我们通过sum求和函数,探究一下x的第一维和第二维的意义?从第8个和第9个输入输出,我们可以看到对于参数axis=0,其结果是数组列的和;而对于参数axis=1,其参数是数组行的和。

对于axis=0第一个维度求和,不是将第一维度(行)中的所有元素相加,而是沿着第一个维度,将对应其他维度(列)的数据相加,分解开来就是第10个输入输出。同理,对于axis=1,是沿着列,将行中的元素相加。

NumPy中对于维度的操作都是以类似这样的逻辑操作的。

多维数组

对于多维数组我们如何准确区分维度呢?下面以图示进行说明:

NumPy中的维度

所以,我的结论就是:在概念上维度是从整体到局部看的,最外围的是第一个维度,然后依次往里,最内部的就是最后一维。

下面我们用代码验证一下上面的结论:

In [19]: x = np.random.randint(0, 9, (2, 3, 4))

In [20]: x
Out[20]:
array([[[0, 7, 5, 5],
        [6, 3, 1, 3],
        [7, 5, 3, 4]],

       [[8, 1, 4, 6],
        [8, 1, 4, 8],
        [3, 0, 8, 2]]])

In [21]: x[0]
Out[21]:
array([[0, 7, 5, 5],
       [6, 3, 1, 3],
       [7, 5, 3, 4]])

In [22]: x[:, 0, :]
Out[22]:
array([[0, 7, 5, 5],
       [8, 1, 4, 6]])

可以看到,第21个输入输出取到的是第一维的第一个元素,第22个输入输出取到的是第二维的第一个元素。大家可以细细体味一下!

原创声明,本文系作者授权云+社区发表,未经许可,不得转载。

如有侵权,请联系 yunjia_community@tencent.com 删除。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • NumPy中的维度Axis

    NumPy中的维度是一个很重要的概念,很多函数的参数都需要给定维度Axis,如何直观的理解维度呢?我们首先以二维数组为例进行说明,然后推广到多维数组。

    卡尔曼和玻尔兹曼谁曼
  • C#中的委托delegate

    委托是一种定义方法签名的类型,可以与具有兼容签名的任何方法关联。你可以通过委托调用方法。委托用于将方法参数传递给其它方法。事件处理程序就是通过委托调用的方法(这...

    卡尔曼和玻尔兹曼谁曼
  • Shapefile属性操作之查

    在前面几篇博文中,我们分别介绍了矢量空间数据的属性数据的增(CREATE)删(DELETE)改(UPDATE)操作,这篇博文我们来聊聊属性数据的查询(Retri...

    卡尔曼和玻尔兹曼谁曼
  • NumPy中的维度Axis

    NumPy中的维度是一个很重要的概念,很多函数的参数都需要给定维度Axis,如何直观的理解维度呢?我们首先以二维数组为例进行说明,然后推广到多维数组。

    卡尔曼和玻尔兹曼谁曼
  • 8.数组

    YungFan
  • java选择排序(最精简代码)

    public class SelectionSort { public void selectionSort(int[] array) { int tem...

    闵开慧
  • PHP开发过程的那些坑(三) ——PHParray_shift函数

    PHP开发过程的那些坑(三)——PHP array_shift函数 (原创内容,转载请注明来源,谢谢) 坑: 最近在自学数据结构,常常用PHP实现各种数据结构...

    用户1327360
  • 关于Go语言中数组的参数传递问题

    一个很微妙的一个问题。从下面的代码可以看出: 1、类型为“[5]int”的数组在作为参数时,其实作为值来传递的。 2、 类型为“[]int”的数组在作为参数...

    李海彬
  • 基础算法(二)

            上一篇:基础算法(一)         1. 冒泡排序(BubbleSort)         原理:依次比较相邻的两个数,将小数放在前面,大数...

    高爽
  • 冒泡排序-排序算法

    mcxfate

扫码关注云+社区

领取腾讯云代金券