如何使用循环访问列表元素的特定部分？

Posted 2023-01-18

我有一个列表是

a = [(0, 0.6249995), (1, 0.12500015), (2, 0.12500016), (3, 0.12500015)]

0,1,2,3是基于LDA建模的主题，而0.6249995、0.12500015、0.12500016和0.12500015是概率。

我只需要访问概率最大的元素，即（0，0.6249995）。

我如何编写一个for循环来做到这一点？

我尝试过这样的事情：

for i in range(len(a[0]) - 1):
    for j in range(len(a)):
    print(max(a[j][1]))

但是这行不通，只是给了我所有的可能性。

我对此可以进行哪些更改？

答案

这是您可以做的（亚历山大·列孔采夫击败我了：]

print(max(a, key=lambda i:i[-1])) #prints element
print(a.index(max(a, key=lambda x:x[-1]))) #prints index of max probability

lambda的作用是max函数根据给定的标准（列表a中每个元素的最后一个索引）求值，并以最大概率返回元组。