将数组的一小部分旋转90度

Question

我想旋转一个数组，但不是整个数组，只是其中的一小部分。

我有一个512X512的数组（基本上是一个以中心点（150,150）和半径为200的高斯圆）。现在我想将数组的一小部分（以中心点（150,150）和半径100为中心）旋转90度。最初我使用了numpy rot90模块，但它旋转了每个数组元素，这不是我想要的。

- physics_for_all

---

你说的数组是指 numpy.array 吗？ - jsalonen

Answer 1

如果您可以使用高级索引描述要旋转的元素，则应该能够使用类似以下内容的方法执行旋转（假设您的数组名为arr）：

arr[rs:re,cs:ce] = np.rot90(np.copy(arr[rs:re,cs:ce]))

这里的rs，re，cs和ce分别表示切片的行起始、行结束、列起始和列结束。

这是为什么需要np.copy调用的示例（至少在numpy 1.3.0中）：

>>> import numpy as np
>>> m = np.array([[i]*4 for i in range(4)])
>>> m
array([[0, 0, 0, 0],
       [1, 1, 1, 1],
       [2, 2, 2, 2],
       [3, 3, 3, 3]])
>>> m[1:3,1:3] = np.rot90(m[1:3,1:3])     # rotate middle 2x2
>>> m
array([[0, 0, 0, 0],
       [1, 1, 2, 1],     # got  1, 2  expected  1, 2  
       [2, 1, 1, 2],     #      1, 1            1, 2
       [3, 3, 3, 3]])

- Andrew Clark

---

我认为你不需要使用 np.copy。 - ecatmur

@ecatmur，也许在更新的版本中不需要，我正在使用1.3.0，所以可能是这个原因。 - Andrew Clark

Answer 2

这里是一些更完整的代码，实现了 F.J. 已经解释过的功能。

这里是代码：

import numpy as np
import scipy

def circle(im, centre_x, centre_y, radius):
    grid_x, grid_y = np.mgrid[0:im.shape[0],0:im.shape[1]]
    return (grid_x-centre_x)**2 + (grid_y-centre_y)**2 < radius**2

centre_x, centre_y, radius = 150, 200, 100
x_slice = slice(centre_x - radius, centre_x + radius)
y_slice = slice(centre_y - radius, centre_y + radius)

im = scipy.misc.imread('1_tree.jpg')
rotated_square = np.rot90(im[x_slice,y_slice].copy())
im[circle(im, centre_x, centre_y,radius)] = rotated_square[circle(rotated_square,
                                                         radius, radius, radius)]
scipy.misc.imsave('sdffs.png',im)

- fraxel