python - 在for循环中索​​引当前值？ Python

Question

所以我有一个 for 循环。

>>> row = [5, 2, 5, 4, 2, 2, 5, 5, 5, 2]
>>> for i in row:
    if i == 5:
        print(row.index(i))
    if i == 2:
        print(row.index(i))

输出

0
1
0
1
1
0
0
0
1

我想得到:0 1 2 4 5 6 7 8 9。 换句话说，我想获取我当前在 for 循环中查看的 i 的索引，如果它是 = 5 或 2 ...任何简单的如何做到这一点？

Answer 1

使用 enumerate() function生成运行索引:

for index, i in enumerate(row):
    if i == 5:
        print(index)
    if i == 2:
        print(index)

或更简单:

for index, i in enumerate(row):
    if i in (2, 5):
        print(index)