我想旋转一个数组,但不是整个数组,只是其中的一小部分。
我有一个512X512的数组(基本上是一个以中心点(150,150)和半径为200的高斯圆)。现在我想将数组的一小部分(以中心点(150,150)和半径100为中心)旋转90度。最初我使用了numpy rot90模块,但它旋转了每个数组元素,这不是我想要的。
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]])
np.copy
。 - ecatmur这里是一些更完整的代码,实现了 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)
numpy.array
吗? - jsalonen