<分区>
这个问题以前可能有人问过,但经过一番搜索后找不到,所以在这里发帖。
假设我有一个数组 A
和一个索引数组 idx
。暂时让两个数组都是二维的。
import numpy as np
A = np.array([[3,3,4],
[4,5,4],
[3,4,5]])
idx = np.array([[1,1],
[2,1],
[1,0],
[0,0]])
现在我想根据idx
中的索引将A
中对应的元素替换为0
。
基本上,我想执行 A[idx]=0
,但这是行不通的。
如何在不运行循环的情况下高效地执行此操作?
最好,建议的解决方案应该可扩展到更高维(3D 及以上)阵列。