是否有更快的方式在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不重复。目前的问题并不反对“转置”功能的稳定性;它要求功能本身。
答案 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
)只会更改数组的.shape
和strides
属性;这是一个观点,而不是副本。
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支持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]])