Python

Question

给定一个列表列表，如

a = [["a","b","c"],
     ["d","e","f"],
     ["g","h","i"]]

如何有效地获取单个列（例如 ["b"、"e"、"h"]）或列的一部分（例如 ["e"、 "h"])?不幸的是我不能切换到 numpy...

Answer 1

这是一个有效的方法：

import operator

def get_column(list_, n):
    return map(operator.itemgetter(n), list_)

然后您可以使用 list() 将输出转换为列表或切片对象。返回的对象是 Python 3 中的地图对象（基本上是生成器）和 Python 中的新列表 2.

Python - 列表矩阵列表的列切片