获取包含元组的列表的索引，其中元组的第一个元素与模式匹配

Question

我有一个元组列表：

    countries = [('Netherlands','31'),
                 ('US','1'),
                 ('Brazil','55'),
                 ('Russia','7')]

现在，我想根据元组中的第一项找到列表的索引。

我试过countries.index('Brazil')，我希望输出是2。但是， returns 一个 ValueError:

ValueError: 'Brazil' is not in list

我知道我可以将此列表转换为 pd DataFrame，然后在第一列中搜索模式匹配。但是，我怀疑有更快的方法可以做到这一点。

Answer 1

您可以使用 enumerate() 来查找您的索引：

尝试：

idx = next(i for i, (v, *_) in enumerate(countries) if v == "Brazil")
print(idx)

打印：

Get index of a list with tuples in which the first element of the tuple matches pattern