Как найти индекс для совпадающих элементов в списке в Python?

Question

У меня есть список имен, и я хочу найти индексы для имен, которые частично соответствуют шаблону. Например, я хочу найти «ohn» и найти индексы всех имен, в которых есть «ohn»>

Вот что я сделал:

r = re.compile(".*ohn")
Index = []
for idx in range(0,len(Names)):
    if r.match(Names[idx]):
        Index.append(idx)
Index

Есть ли лучший способ сделать это?

Answer 1

Вы можете использовать enumerate для генерации индексов для последовательности в понимании списка:

[idx for idx, name in enumerate(Names) if r.match(name)]