6
NumPy通过使用索引列表为每行选择特定的列索引
我正在努力选择NumPy矩阵每行的特定列。 假设我有以下矩阵,我将其称为X: [1, 2, 3] [4, 5, 6] [7, 8, 9] 我list每行还有一个列索引,我将其称为Y: [1, 0, 2] 我需要获取值: [2] [4] [9] 除了使用listwith索引之外Y,我还可以生成形状与X每个列都是0-1值范围内的bool/的形状相同的矩阵int,指示这是否是必需的列。 [0, 1, 0] [1, 0, 0] [0, 0, 1] 我知道这可以通过遍历数组并选择所需的列值来完成。但是,这将在大数据数组上频繁执行,这就是为什么它必须尽可能快地运行。 因此,我想知道是否有更好的解决方案? 谢谢。