Numpy将函数应用于矩阵

Question

我有两个随机索引生成器，从0到3

idx_1 = np.random.choice(4, 2, replace=False)
idx_2 = np.setdiff1d(range(4), idx_1)

我有一个3D numpy数组，example_array，由4个矩阵组成，2个函数，比如func1和func2。

我想做点什么

example_array[idx1[0]] = func1(example_array[idx1[0]])
example_array[idx1[1]] = func1(example_array[idx1[1]])
example_array[idx2[0]] = func2(example_array[idx2[0]])
example_array[idx2[1]] = func2(example_array[idx1[0]])

这样做的快速方法是什么，没有明确地如上所述？

Answer 1

如何对func1()和func2()进行矢量化以便

example_array[idx1] = func1(example_array[idx1])
example_array[idx2] = func2(example_array[idx2])