gpt4 book ai didi

python - Numpy 将函数应用于选定的矩阵索引

转载 作者:太空宇宙 更新时间:2023-11-03 15:14:28 24 4
gpt4 key购买 nike

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

如果没有像上面那样明确地编写它们,有什么快速的方法可以做到这一点?

最佳答案

func1()func2() 进行向量化怎么样?

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

关于python - Numpy 将函数应用于选定的矩阵索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43980258/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com