我喜欢 python 处理变量交换的方式:
a, b, = b, a
我也想使用这个功能在数组之间交换值,不仅是一次一个,而且是多个(不使用临时变量)。这不是我所期望的(我希望沿第三维的两个条目都可以交换):
import numpy as np
a = np.random.randint(0, 10, (2, 3,3))
b = np.random.randint(0, 10, (2, 5,5))
# display before
a[:,0, 0]
b[:,0,0]
a[:,0,0], b[:, 0, 0] = b[:, 0, 0], a[:,0,0] #swap
# display after
a[:,0, 0]
b[:,0,0]
有人有想法吗?当然,我总是可以引入一个额外的变量,但我想知道是否有更优雅的方法来做到这一点。