【问题标题】:Select columns of a matrix out of rows indexes contained in an array [duplicate]从数组中包含的行索引中选择矩阵的列[重复]
【发布时间】:2020-06-23 22:22:09
【问题描述】:

假设我有一个给定维度的矩阵,以及一个包含一系列值的数组。数组中包含的值是我要从矩阵中提取值的行的索引。所以,一个例子可能是:

A = np.array([[3, 6, 7, 5, -3, 0], [5, -2, 2, 51, -13, 8], [13, -17, 18, 22, -12, 90], [33, -12, 12, 32, -42, 90]])

B = np.array([0, 2, 3])

预期的结果:

array([[ 3,  7,  5],
       [ 5,  2, 51],
       [13, 18, 22],
       [33, 12, 32]])

【问题讨论】:

标签: python numpy


【解决方案1】:

这在numpy 中非常简单——只需执行:

A[:, B]

一些有用的参考:https://docs.scipy.org/doc/numpy/user/basics.indexing.html

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2018-06-28
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2018-06-16
    • 2011-05-27
    相关资源
    最近更新 更多