我有两个随机索引生成器,从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]])
这样做的快速方法是什么,没有明确地如上所述?
答案 0 :(得分:1)
如何对func1()
和func2()
进行矢量化以便
example_array[idx1] = func1(example_array[idx1])
example_array[idx2] = func2(example_array[idx2])