给出一个列表,如
a = [["a","b","c"],
["d","e","f"],
["g","h","i"]]
如何有效地获得单个列(例如[" b"," e"," h"])或列的一个片段(例如[" E"" H"])?不幸的是,我无法转向numpy ......
答案 0 :(得分:2)
这是一种有效的方式:
import operator
def get_column(list_, n):
return map(operator.itemgetter(n), list_)
然后,您可以使用list()
将输出转换为列表或切片对象。返回的对象是Python 3中的地图对象(基本上是生成器)和Python 2中的新列表。