Skip to main content
 首页 » 编程设计

list之获取矩阵某些行的 Pythonic 方法

2025年05月04日10insus

我在想我几年前用 Python 编写的代码,在某些时候它必须通过索引获取列表列表的一些元素。

我记得我做过这样的事情:

def getRows(m, row_indices): 
    tmp = [] 
    for i in row_indices: 
        tmp.append(m[i]) 
    return tmp 

现在我从那时起学到了更多一点,我会使用这样的列表推导式:

[m[i] for i in row_indices] 

但我仍然想知道是否有更 pythonic 的方法来做到这一点。有什么想法吗?

我还想知道 numpy 或任何其他数组库的替代方案。

请您参考如下方法:

值得一看NumPy用于其切片语法。在链接页面中向下滚动,直到您到达“索引、切片和迭代”。