我在想我几年前用 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用于其切片语法。在链接页面中向下滚动,直到您到达“索引、切片和迭代”。
