Pythonで指定したインデックスリストの要素をまとめて取得する方法

Pythonでは、リストの個々の要素へアクセスする際に、角括弧 [] とインデックス番号を使用します。しかし、複数のインデックスをまとめて指定して要素を取り出したい場合には、この方法をそのまま使うことができません。そこで、以下に挙げるアプローチを活用します。

方法1：2つのリストを使う

元のリストとは別に、取得したいインデックスをもう1つのリストとして用意します。そして、リスト内包表記（forループ）でインデックスのリストを順に処理し、その値を元のリストに渡すことで、対応する要素を取り出します。

サンプルコード

given_list = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri']
index_list = [1, 3, 4]

# リストの内容を出力
print('Given list : ' + str(given_list))
print('List of Indices : ' + str(index_list))

# リスト内包表記で要素を取得
res = [given_list[n] for n in index_list]

# 結果を表示
print('Result list : ' + str(res))

実行結果

上記のコードを実行すると、次の出力が得られます。

Given list : ['Mon', 'Tue', 'Wed', 'Thu', 'Fri']
List of Indices : [1, 3, 4]
Result list : ['Tue', 'Thu', 'Fri']

方法2：mapと__getitem__を使う

forループの代わりに、map関数とリストの特殊メソッドである__getitem__を組み合わせても、同じ結果が得られます。__getitem__は要素へのアクセスを担当するメソッドで、これをmapに渡すことで、インデックスリストの各要素に対して一括でアクセスできます。

サンプルコード

given_list = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri']
index_list = [1, 3, 4]

# リストの内容を出力
print('Given list : ' + str(given_list))
print('List of Indices : ' + str(index_list))

# mapと__getitem__で要素を取得
res = list(map(given_list.__getitem__, index_list))

# 結果を表示
print('Result list : ' + str(res))

実行結果

上記のコードを実行すると、次の出力が得られます。

Given list : ['Mon', 'Tue', 'Wed', 'Thu', 'Fri']
List of Indices : [1, 3, 4]
Result list : ['Tue', 'Thu', 'Fri']

補足：operator.itemgetterを使う方法

標準ライブラリのoperatorモジュールに含まれるitemgetterを使うと、複数のインデックスを一度に指定して要素を取得できます。コードが簡潔になるため、覚えておくと便利です。

サンプルコード

from operator import itemgetter

given_list = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri']
index_list = [1, 3, 4]

# 複数のインデックスを一度に指定して要素を取得
res = list(itemgetter(*index_list)(given_list))

# 結果を表示
print('Result list : ' + str(res))

実行結果

Result list : ['Tue', 'Thu', 'Fri']

まとめ

いずれの方法でも同じ結果が得られます。可読性を重視するならリスト内包表記、関数型のスタイルで処理したい場合はmapと__getitem__の組み合わせ、簡潔さを求めるならoperator.itemgetterがおすすめです。用途に応じて使い分けましょう。