Numpy将函数应用于矩阵

时间:2017-05-15 13:01:21

标签: python arrays function numpy

我有两个随机索引生成器,从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]])

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

1 个答案:

答案 0 :(得分:1)

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

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