从Python列表列表的内部列表中获取第n个元素

可以说我有:

a = [ [1,2], [2,9], [3,7] ]

我想检索每个内部列表的第一个元素:

b = [1,2,3]

无需执行此操作(我当前的hack):

for inner in a:

b.append(inner[0])

我敢肯定有一个班轮,但我真的不知道我在寻找什么。

回答:

只需将您的列表组合更改为:

b = [el[0] for el in a]

要么:

from operator import itemgetter

b = map(itemgetter(0), a)

或者,如果您要处理“适当的数组”:

import numpy as np

a = [ [1,2], [2,9], [3,7] ]

na = np.array(a)

print na[:,0]

# array([1, 2, 3])

zip

print zip(*a)[0]

以上是 从Python列表列表的内部列表中获取第n个元素 的全部内容, 来源链接: utcz.com/qa/401461.html

回到顶部