列表矩阵列表的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()将输出转换为列表，或者对对象进行切片。返回的对象是python3中的一个map对象（基本上是一个生成器）和python2中的一个新列表。你知道吗