理解numpy的rollaxis与swapaxes函数

现在再说rollaxis函数,这个函数有三个参数 

numpy.rollaxis(arr, axis, start)

arr:输入数组
axis:要向后滚动的轴,其它轴的相对位置不会改变
start:默认为零,表示完整的滚动。会滚动到特定位置

import numpy as np
 
a = np.arange(8).reshape(2, 2, 2)
 
print '原数组:'
print a
print '\n'
# 将轴 2 滚动到轴 0(宽度到深度)
 
print '调用 rollaxis 函数:'
print np.rollaxis(a, 2)
# 将轴 0 滚动到轴 1:(宽度到高度)
print '\n'
 
print '调用 rollaxis 函数:'
print np.rollaxis(a, 2, 1)
原数组:
[[[0 1]
  [2 3]]
 
 [[4 5]
  [6 7]]]
 
 
调用 rollaxis 函数:
[[[0 2]
  [4 6]]
 
 [[1 3]
  [5 7]]]
 
 
调用 rollaxis 函数:
[[[0 2]
  [1 3]]
 
 [[4 6]
  [5 7]]]

程序运行np.rollaxis(a, 2)时,讲轴2滚动到了轴0前面,其他轴相对2轴位置不变(start默认0),数组下标排序由0,1,2变成了1,2,0

这时数组按下标顺序重组,例如第一个数组中[0,1]下标为[000,001],其中0的下标变动不影响值,1位置的下标由001变成010,第一位的下标滚动到最后一位下标的后面,值由1(001)变成2(010):

可以得出轴的滚动就是下标的滚动,同理,运行np.rollaxis(a, 2, 1)时将下标0,1,2变为0,2,

numpy.swapaxes(arr, axis1, axis2)函数则是交换,将axis1和axis2进行互换

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值