在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]

我已经将这个问题标记为重复，the dupe post的答案有更多细节。

Answer 2

我看到你正在使用numpy数组。在这种情况下，您也可以这样做：

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