listで一部該当する要素のインデックスを取得したい

Question

[やりたいこと]
特定の文字列を含むドメインをリストから検索→インデックスを取得したい。

listに該当の要素が含まれているかどうかまでは書いたのですが、ここから該当する要素のインデックスを取得するにはどのように書くといいでしょうか。ご教授いただければと思います。

domain = ['aaa.com', 'bbb.com', 'ccc.net']

search_domain = 'aaa'

for d in domain:
    if search_domain in d:
        print('hit')

Answer 1

完全に一致するものであればリストの.index()が使えたのですが、今回は別の方法が必要なようです。enumerate を使ってみては。

idx = [idx for idx, d in enumerate(domain)
       if search_domain in d]

Comments

[teosu]

ご回答ありがとうございます！

Answer 2

enumerateを使えば実現可能です。

domain = ['aaa.com', 'bbb.com', 'ccc.net']

search_domain = 'aaa'

for i,d in enumerate(domain):
    if search_domain in d:
        print(f'hit : {i}')

あ〜、解決後ですが…(^ ^;

私ならこういう風にリストを抽出してしまいますかね…

domains = [
    'aaa.com'
    ,'bbb.com'
    ,'ccc.net'
    ,'aaa.co.jp'
]

search_domain = 'aaa'

target_domains = list(filter(lambda s: search_domain in s,domains))

# ↓内包表記バージョン。↑か↓のお好きな方をどうぞ。
#target_domains = [domain for domain in domains if search_domain in domain]

Comments

[teosu]

迅速なご回答ありがとうございます！enumerateをもっとしっかり勉強します。ありがとうございました！

[teosu]

追加のご回答ありがとうございます。lambdaを使った方法も勉強します。ありがとうございます！

Answer 3

一番単純なのはインデックスを示すカウンターを別に定義する方法です。
コードは以下の通りです。

domain = ['aaa.com', 'bbb.com', 'ccc.net']

idx = 0   # インデックスの初期設定
search_domain = 'aaa'

for d in domain:
    if search_domain in d:
        print('hit')
        print(idx)   # hitしたインデックスの出力

    idx = idx + 1    # インデックスの更新

Comments

[teosu]

カウンターを別定義する方法は抜けていました。こちらの方法でも書き直してみます。回答ありがとうございます！