翻转并旋转numpy数组

时间:2016-06-28 20:46:27

标签: python arrays numpy

是否有更快的方式在numpy中翻转和旋转数组?例如,顺时针旋转一次然后翻转?

import numpy as np
a = np.arange(0,10)
b = np.arange(-11,-1)

ar = np.array([a,b])

print ar
print ar.shape

ar = np.rot90(ar, 3)
print np.fliplr(ar)
print ar.shape

输出:

[[  0   1   2   3   4   5   6   7   8   9]
 [-11 -10  -9  -8  -7  -6  -5  -4  -3  -2]]
(2, 10)

[[  0 -11]
 [  1 -10]
 [  2  -9]
 [  3  -8]
 [  4  -7]
 [  5  -6]
 [  6  -5]
 [  7  -4]
 [  8  -3]
 [  9  -2]]
(10, 2)
[Finished in 0.1s]

P.S。:这个问题与Transposing a NumPy array不重复。目前的问题并不反对“转置”功能的稳定性;它要求功能本身。

3 个答案:

答案 0 :(得分:2)

np.rot90的代码,k=3的代码:

    # k == 3
    return fliplr(m.swapaxes(0, 1))

所以

In [789]: np.fliplr(ar.swapaxes(0, 1))
Out[789]: 
array([[-11,   0],
     ...
       [ -3,   8],
       [ -2,   9]])

所以你的

fliplr(rot90(ar, 3))

变为

 np.fliplf(np.fliplr(ar.swapaxes(0, 1)))
 # the flips cancel
 ar.swapaxes(0,1)
 # but this is just
 ar.T

所以你的一对动作会减少转置。

transpose(以及swap)只会更改数组的.shapestrides属性;这是一个观点,而不是副本。

np.fliplr还会创建一个视图,使用[:,::-1]改变步幅。

原始ar

In [818]: ar
Out[818]: 
array([[  0,   1,   2,   3,   4,   5,   6,   7,   8,   9],
       [-11, -10,  -9,  -8,  -7,  -6,  -5,  -4,  -3,  -2]])

In [819]: x=np.fliplr(np.rot90(ar,3))  # your pair of actions

In [820]: x
Out[820]: 
array([[  0, -11],
       [  1, -10],
         ...
       [  8,  -3],
       [  9,  -2]])

In [821]: x[0,1]=11

In [822]: x
Out[822]: 
array([[  0,  11],
       [  1, -10],
        ...
       [  9,  -2]])

In [823]: ar
Out[823]: 
array([[  0,   1,   2,   3,   4,   5,   6,   7,   8,   9],
       [ 11, -10,  -9,  -8,  -7,  -6,  -5,  -4,  -3,  -2]])

更改x的值会更改ar的值。尽管使用了2个函数,x仍然是view的{​​{1}}。

不需要2个功能,但它们也不贵。我们正在谈论微秒v纳秒的时间。 (我在Ipython中的ar次比你的小得多)

timeit

答案 1 :(得分:1)

翻转和旋转(基于您的示例)是matrix transpose:矩阵转置是矩阵维度的排列:例如,第一维度变为第二维度,反之亦然。

支持numpy.transpose功能:

numpy.transpose(a, axes=None)
     

置换数组的尺寸。

     

<强>参数

     
      
  • a : array_like:输入数组。
  •   
  • axes:整数列表,可选   默认情况下,反转尺寸,否则根据给定的值对轴进行置换。
  •   
     

<强>返回

     
      
  • p : ndarray:a轴被置换。尽可能返回视图。
  •   

答案 2 :(得分:1)

那将是transpose

>>> import numpy as np
>>> a = np.arange(0,10)
>>> b = np.arange(-11,-1)
>>> ar = np.array([a,b])
>>> ar.T
array([[  0, -11],
       [  1, -10],
       [  2,  -9],
       [  3,  -8],
       [  4,  -7],
       [  5,  -6],
       [  6,  -5],
       [  7,  -4],
       [  8,  -3],
       [  9,  -2]])
>>> np.transpose(ar)
array([[  0, -11],
       [  1, -10],
       [  2,  -9],
       [  3,  -8],
       [  4,  -7],
       [  5,  -6],
       [  6,  -5],
       [  7,  -4],
       [  8,  -3],
       [  9,  -2]])