从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 itemgetterb = map(itemgetter(0), a)
或者,如果您要处理“适当的数组”:
import numpy as npa = [ [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