numpy - 使用另一个 numpy 数组索引 numpy 数组

Question

我有一个 2D numpy 数组和另一个 2D numpy 数组，其中包含我想从原始矩阵访问的特定列索引。我怎样才能有效地做到这一点？

例如，我有数组

A = np.array([[4 1 4 5 2 1 2 1 2 5],
 [4 5 5 3 3 2 5 5 3 2],
 [1 3 5 5 3 5 3 5 4 5]])

和列索引数组（与 A 具有相同的行数）

ind = np.array([[7 3 6 0 4],
 [2 5 7 3 8],
 [7 0 3 6 4]])

现在，我可以访问与 ind 对应的 A 的元素，这样我就可以执行 A[ind] = 0，这样第 1 行中的第 7、3、6、0、4 列是 0，第 2、5、7、3 列第 2 行中的 ,8 为 0 等。

标签： numpyindexing