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中的新列表。