Python - 列表矩阵

时间:2016-03-25 10:04:47

标签: python list slice

给出一个列表,如

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

如何有效地获得单个列(例如[" b"," e"," h"])或列的一个片段(例如[" E"" H"])?不幸的是,我无法转向numpy ......

1 个答案:

答案 0 :(得分:2)

这是一种有效的方式:

import operator

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

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