3

我正在尝试使用数组切片来反转NumPy数组的一部分。例如,如果我的数组是

a = np.array([1,2,3,4,5,6])

然后我可以得到一片 b

b = a[::-1]

这是原始数组的视图。我想要的是一个部分反转的视图,例如

1,4,3,2,5,6

如果您不完全按照 NumPy 的设计方式进行操作,我会遇到 NumPy 的性能问题,所以如果可能的话,我想避免“花哨”的索引。

4

3 回答 3

8

如果您不喜欢按一个索引关闭

>>> a = np.array([1,2,3,4,5,6])
>>> a[1:4] = a[1:4][::-1]
>>> a
array([1, 4, 3, 2, 5, 6])
于 2013-07-04T11:36:50.193 回答
4
>>> a = np.array([1,2,3,4,5,6])
>>> a[1:4] = a[3:0:-1]
>>> a
array([1, 4, 3, 2, 5, 6])
于 2013-07-04T11:28:06.880 回答
0

您可以使用排列矩阵(这是部分反转数组的最简单的方法)。

a = np.array([1,2,3,4,5,6])
new_order_for_index = [1,4,3,2,5,6] # Careful: index from 1 to n !

# Permutation matrix
m = np.zeros( (len(a),len(a)) )
for index , new_index  in enumerate(new_order_for_index ):
    m[index ,new_index -1] = 1

print np.dot(m,a)
# np.array([1,4,3,2,5,6])
于 2013-07-04T11:34:00.400 回答