python - 根据特定索引处的嵌套列表中的元素过滤列表

Question

我有一个包含列表的列表:

[['4.2','3.4','G'],['2.4','1.2','H'],['8.7','5.4','G']]

我想通过引用列表列表中每个列表的第三部分中的字母来从列表列表中获取值。

例如，我希望 python 为列表列表中的每个项目打印元素 由字母 'G' 表示。

output = [4.2,3.4]
         [8.7,5.4]

这是我尝试过的:

L = [['4.2','3.4','G'],['2.4','1.2','H'],['8.7','5.4','G']]
newList = []

for line in L:
    if line[0][2] == 'G'
        newList.append([float(i) for i in line[0:2]])
print(newList)

我的错误将出现在第 5 行，因为我不确定我是否能够这样做。问候。

Answer 1

简单的列表理解:

L = [['4.2','3.4','G'],['2.4','1.2','H'],['8.7','5.4','G']]
newList = [l[0:2] for l in L if l[2] == 'G']

print(newList)

输出:

[['4.2', '3.4'], ['8.7', '5.4']]