Python中发现列表中的一个项目的索引，其中一些列并不重要

Question

我有一个列表在Python这就像下面的例子：a = [[1,2,'aaa'] , [3,4,'nnnn']

我怎么可以随便说[1,2，*]其中*表示并不重要，得到[1,2，“AAA”]的指数？换句话说，我希望获得项目的指标，他们的第一列是1和他们的第二列和第三列的值并不重要。

Answer 1

你可以使用extended iterable unpacking：

a = [[1, 2, 'aaa'], [3, 4, 'nnnn']]

indices = [i for i, (first, second, *_) in enumerate(a) if (first, second) == (1, 2)]
print(indices)

产量

[0]

Answer 2

您可以使用operator.itemgetter一个可扩展解决方案：

from operator import itemgetter

a = [[1,2,'aaa'] , [3,4,'nnnn']]
getter = itemgetter(0, 1)  # get first and second items
indices = [idx for idx, item in enumerate(a) if getter(item) == (1, 2)]  # [0]

Answer 3

如何关于一个简单明了的循环？

a = [[1,2,'aaa'] , [3,4,'nnnn']
for i, item in enumerate(a):
    if item[0] == 1 and item[1] == 2:
    print(i)
    break