从两个numpy数组中随机选择元素

Question

我正在尝试通过从每个数组中随机选择元素来将两个 numpy 数组合并在一起。 假设我有两个长度相等的数组x 和y，如下所示：

x = np.arange(10)
y = np.arange(10, 20)

还有一个面具r：

r = np.random.choice([True, False], 10)

那么有什么方法可以从x 中选择元素，其中r 是True，从y 中选择元素，其中r 是False？

我不必使用掩码方法，但我需要一些快速的东西，因为 x 和 y 实际上会比 10 长得多，所以理想情况下不涉及循环。

Answer 1

这个怎么样？

z = y.copy()
z[r] = x[r]

Answer 2

这是使用np.where 的np.where(<condition>, <where_true>, <where_false>) 语法的单行代码：

z = np.where(r, x, y)