如何使用另一个包含索引的 2D numpy 数组对 2D numpy 数组进行切片?

问题描述 投票:0回答:0

我有一个大小为 (3000,3000) 的稀疏矩阵 A,还有另一个大小为 (83068, 2) 的矩阵 B,其中包含 A 的非零元素的索引。

是否有可能得到一个二维矩阵 C,使其仅包含 A 中的非零元素?

这是我迄今为止尝试过的:

A[B]

但结果大小是 (83068, 2, 3000)

python arrays numpy indexing slice
© www.soinside.com 2019 - 2024. All rights reserved.