如何从python列表中的每个列表中拉出第一个元素?

我有一个带有嵌套列表的列表,该嵌套列表内有15980个列表的列表,该列表由两个数字组成(这是pytorch模型的概率)。我需要迭代从每个列表中解压缩第一个数字,并进行一系列序列附加到数据框。我尝试了[x for l in list for x in l],但是那只是去除了第二个嵌套,并在其中保留了15980个列表。

示例:

a = [.22,.33]
b = [.23,.34]
c = [.76,.14]

d = [a,b,c]
e = [d]

e是我正在使用的。我需要将a,b和c中的第一个数字放入序列中。

gzblue_zhuzh 回答:如何从python列表中的每个列表中拉出第一个元素?

一种简单的理解方法:

[x[0] for x in d for d in e]

计算结果为

[0.22,0.23,0.76]
本文链接:https://www.f2er.com/3162899.html

大家都在问