在numpy数组中交换两个值.

Question

有没有比以下代码更有效的东西来交换numpy 1D数组的两个值？

input_seq = arange(64)

ix1 = randint(len(input_seq))
ixs2 = randint(len(input_seq))

temp = input_seq[ix2]
input_seq[ix2] = input_seq[ix1] 
input_seq[ix1] = temp

Answer 1

你可以使用元组解包.元组解包允许您避免在代码中使用临时变量(实际上我相信Python代码本身在幕后使用临时变量,但它的级别要低得多,因此速度要快得多).

input_seq[ix1], input_seq[ix2] = input_seq[ix2], input_seq[ix1]

我已经将这个问题标记为重复,在dupe帖子中的答案有更多细节.

Answer 2

我看到你正在使用numpy数组.在这种情况下,您也可以这样做:

input_seq[[ix1, ix2]] = input_seq[[ix2, ix1]]