Python – リスト（マトリックス）内の特定の文字列Kを含む行を見つける方法

Pythonでは、行列（リストのリスト）の中から特定の文字列「K」を含む行を見つけたい場面があります。そのような場合、「enumerate」関数によるインデックス付きの反復処理と、「append」メソッドを組み合わせることで、シンプルかつ効率的に実装できます。

コード例

以下に具体的な実装例を示します。

my_list = [["Pyt", "fun", "python"], ["python", "rock"],["Pyt", "for", "CS"], ["Keep", "learning"]]

print("The list is :")
print(my_list)

K = "Pyt"

my_result = []

for idx, element in enumerate(my_list):

    if K in element:
        my_result.append(idx)

print("The result is :")
print(my_result)

出力

The list is :
[['Pyt', 'fun', 'python'], ['python', 'rock'], ['Pyt', 'for', 'CS'], ['Keep', 'learning']]
The result is :
[0, 2]

解説

• まず、複数の文字列を要素として持つネストされたリスト（行列）を定義し、コンソールに表示します。

• 検索対象となる文字列「K」の値を定義します。ここでは「Pyt」としています。

• 結果を格納するための空のリストを作成します。

• 「enumerate」関数を使ってリストを反復処理することで、各要素（行）とそのインデックスを同時に取得できます。

• if文による条件分岐で、各行の中に「K」が含まれているかどうかを判定します。「in」演算子を使えば、リスト内に指定した文字列が存在するかどうかを簡単に確認できます。

• 条件を満たす場合、その行のインデックスを結果用のリストに追加します。

• 最後に、結果をコンソールに出力して完了です。この例では、0番目と2番目の行に「Pyt」が含まれているため、[0, 2]が出力されています。

補足：リスト内包表記を使ったより簡潔な書き方

上記のコードは、リスト内包表記を使うことでさらに簡潔に書くこともできます。

my_result = [idx for idx, element in enumerate(my_list) if K in element]

このように1行で同じ結果を得られるため、Pythonらしい書き方として覚えておくと便利です。可読性と処理効率の両面で優れており、実務でも広く使われています。