Numpy运算中的axis参数设置

最近在学习Numpy时,对函数设置中axis(轴)参数有些困惑,学习了这两篇文章:
Python · numpy · axis
Numpy:对Axis的理解
感觉自己弄明白了,在这里分享我的理解,欢迎交流和指正。

举例说明更明白,那就直接上例子吧。
数组示例
对于图中所示的数组,shape为(2,3,4),表示这是一个三维数组,每个维度的大小分别是2,3,4,换言之,就是在相应的维度上有2/3/4个元素。以第一维维度(对应axis=0)为例,其大小为2,也就是有两个元素,即图中的两个 3 × 4 3\times4 3×4数组。

axis(轴)操作是依据什么规则呢?一般来说,对于N维数据,axis的数字从0到N-1,对应于数组从外到内层(看括号方向)。
我的理解:换个角度,我们可以想象在多维空间进行操作。选取某根坐标轴(axis),相应得到多个切面,然后对这多个切面进行操作。 我们可以在每个维度取出一个“元素”来看看(对应于空间的一个“切面”):
某一维的元素示例
观察结果我们发现,某一维的元素,其大小正好是其余两维的大小。例如对于上面shape为(2,3,4)的数组b,第一维(axis=0)的元素b[1,:,:]的shape为(3,4),正是其余两维的大小。第二维(axis=1)、第三维(axis=2)的情况也是如此。

现在我们来看np.sum()操作,它是把某个方向上的元素相加,对应到空间中,相当于多个切面叠加到一起,合而为一。因此得到的结果的shape与元素一致。

np.sum()示例

np.sort()更有意思,它的结果也体现轴操作的特点。比如axis=0时,注意看是数组b中的[1,2,3,4]和[3,2,5,6]对应逐元素排序,[5,2,4,1]和[1,3,6,7]对应逐元素排序,[3,1,1,2]和[6,9,3,1]对应逐元素排序。
np.sort()示例

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: numpy的diff函数是用来计算数组元素之间的差值的,也就是说,它可以计算数组相邻元素的差值。具体而言,如果输入一个一维数组a,那么diff(a)将返回一个一维数组b,其b[i] = a[i+1] - a[i]。如果输入一个二维数组a,那么可以指定沿着哪个轴计算差值。 下面是一个使用numpy的diff函数的示例代码: ```python import numpy as np a = np.array([1, 3, 6, 10, 15]) b = np.diff(a) print(b) # 输出 [2 3 4 5] c = np.array([[1, 2, 3], [4, 5, 6]]) d = np.diff(c, axis=0) print(d) # 输出 [[3 3 3]] ``` 在第一个示例,我们计算了数组a相邻元素的差值,并将结果存储在数组b。在第二个示例,我们计算了二维数组c沿着第0个轴(即行)的差值,并将结果存储在数组d。注意,这里的结果是一个一维数组,因为我们计算的是每一行之间的差值。 ### 回答2: numpy的diff函数是用于计算数组元素之间的差值。它可以计算出数组每个元素与其相邻元素之间的差值。例如,对于数组[1, 3, 5, 9],使用diff函数后将返回 [2, 2, 4]。默认情况下,diff函数将计算相邻元素之间的差值,但也可以通过指定参数来计算不同步长的差值。 在使用diff函数时,可以通过指定参数n来设置差值的步长。例如,对于数组[1, 2, 4, 7, 11],如果设置n为2,则diff函数将返回[3, 5],表示数组每隔2个元素的差值。 此外,diff函数还可以用于多维数组的计算。对于多维数组,可以通过指定参数axis来指定不同的维度进行差值计算。例如,对于二维数组[[1, 2, 3], [4, 6, 8]],可以使用axis=0来计算每一列之间的差值,使用axis=1来计算每一行之间的差值。 此外,diff函数还可以通过指定参数prepend和append来在结果数组的前部和尾部添加指定的元素。这些参数可以是一个数值或一个数组。使用这些参数可以方便地在计算差值之后,将其结果与原数组进行拼接。 总的来说,numpy的diff函数是一个方便的计算数组差值的工具,它可以用于一维或多维数组,并且支持不同步长的计算,并且可以通过设置参数prepend和append来添加额外的元素。 ### 回答3: numpy库是一个开源的数值计算库,是Python语言非常重要的扩展库之一。其的diff函数是numpy库提供的一个用于计算数组元素之间的差值的函数。 在numpy,diff函数可以用于计算数组相邻元素之间的差值。它的语法形式为:numpy.diff(a, n=1, axis=-1)。 其,a表示输入的数组;n表示计算差值的次数,默认为1;axis表示沿着哪个轴计算差值,默认为-1,即最后一个轴。 该函数将会返回一个由相邻元素之间的差值构成的新数组。例如,若输入数组a为[1, 2, 4, 7, 0],调用diff函数后会返回一个新数组[-1, 2, 3, -7],这是因为2-1=1,4-2=2,7-4=3,0-7=-7。 diff函数还可以进行多次差值计算。例如,若将n参数设置为2,那么将会对一阶的差值结果进行二阶差值运算。如果将n设置为负数,则表示执行逆差值,即还原原来的数组。 此外,diff函数还可以通过axis参数来指定沿着哪个轴计算差值。默认情况下,它沿着最后一个轴进行计算。可以通过axis参数传入一个整数或元组来指定其他轴。 总之,numpy的diff函数是一个非常有用的功能,可以方便地计算数组相邻元素之间的差值,用于快速的计算数值序列的变化趋势等任务。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值