将 function 应用于 Numpy 阵列的特定行

Question

假设我有一个二维数组，一个 function 和一个特定行的“掩码”，如下所示：

my_array = np.array([[0,1],[2,3],[4,5],[6,7]])
my_mask = np.array([0,1,0,1])
my_func = lambda x: x * 2

如何将此 function 应用于掩码中为真的数组行？ 即对于上面的示例，结果将是：

array([[0,1],[4,6],[4,5],[12,14]])

Answer 1

您可以使用 boolean 索引：

mask = my_mask==1
my_array[mask] = my_func(my_array[mask])

Output：

array([[ 0,  1],
       [ 4,  6],
       [ 4,  5],
       [12, 14]])