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 中是一个 map 对象(基本上是一个生成器)，在 Python 2 中是一个新列表。