Pythonで文字列リストからK番目の単語の文字位置を取得する方法

文字列のリストから「K」番目の単語の文字位置を求めたい場合、リスト内包表記とenumerate関数を組み合わせることで、簡潔に実装できます。この記事では、その具体的な方法をサンプルコードとともに解説します。

サンプルコード

以下に実際のプログラム例を示します。

my_list = ["python", "is", "fun", "to", "learn"]

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

K = 15
print("The value of K is :")
print(K)

my_result = [element[0] for sub in enumerate(my_list) for element in enumerate(sub[1])]

my_result = my_result[K]

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

実行結果

The list is :
['python', 'is', 'fun', 'to', 'learn']
The value of K is :
15
The result is :
2

処理の流れと解説

• まず、文字列のリストを定義し、コンソールに表示します。

• 次に、取得したい位置を示す値「K」を定義し、こちらもコンソールに表示します。

• リスト内包表記の中でenumerateを使用し、リストの各要素（単語）に対して反復処理を行います。

• さらに内側のenumerateによって、各単語に含まれる文字ごとのインデックス（0番目の要素）へアクセスします。

• これらの値はリストとしてまとめられます。

• enumerateを使用しているため、生成される各要素は整数型のインデックスになります。

• 最後に、作成したリストの「K」番目の要素を取り出し、結果としてコンソールに表示します。

ポイント

この手法では、リスト全体を1次元のインデックス空間として扱えるのが特徴です。つまり、「python」「is」「fun」といった複数の単語にまたがって通し番号で文字位置を数えられるため、K番目の文字がどの単語の何文字目に相当するかを簡単に特定できます。ネストされたループを書く必要がなく、リスト内包表記だけで完結する点も、コードを読みやすく保つ大きなメリットです。