我正在尝试使用数组切片来反转NumPy数组的一部分。例如,如果我的数组是
a = np.array([1,2,3,4,5,6])
然后我可以得到一片 b
b = a[::-1]
这是原始数组的视图。我想要的是一个部分反转的视图,例如
1,4,3,2,5,6
如果您不完全按照 NumPy 的设计方式进行操作,我会遇到 NumPy 的性能问题,所以如果可能的话,我想避免“花哨”的索引。
如果您不喜欢按一个索引关闭
>>> a = np.array([1,2,3,4,5,6])
>>> a[1:4] = a[1:4][::-1]
>>> a
array([1, 4, 3, 2, 5, 6])
>>> a = np.array([1,2,3,4,5,6])
>>> a[1:4] = a[3:0:-1]
>>> a
array([1, 4, 3, 2, 5, 6])
您可以使用排列矩阵(这是部分反转数组的最简单的方法)。
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])