PythonのバイナリリストからTrue値のインデックスを効率的に取得する方法

Pythonのリストに True / False や 0 / 1 のような値だけが含まれている場合、そのリストは「バイナリリスト」と呼ばれます。本記事では、バイナリリストを対象に、要素がTrueと評価される位置（インデックス）を抽出する方法を2つ紹介します。

方法1：enumerate関数を使う

enumerate() 関数を使うと、リストの各要素とそのインデックスを同時に取り出せます。内包表記の中で条件判定を行い、Trueと評価される要素のインデックスだけを新しいリストに格納します。Pythonでは 1 も True として扱われるため、数値が混在するリストでも正しく動作します。

コード例

listA = [True, False, 1, False, 0, True]
# 元のリストを表示
print("元のリスト :\n ", listA)
# enumerate() を使用
res = [i for i, val in enumerate(listA) if val]
# 結果を表示
print("True値を持つインデックス :\n ", res)

実行結果

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

元のリスト :
[True, False, 1, False, 0, True]
True値を持つインデックス :
[0, 2, 5]

方法2：itertools.compressを使う

itertools モジュールの compress() 関数を使う方法もあります。compress(データ, セレクタ) の形式で呼び出し、セレクタ側の値がTrueである位置に対応するデータ側の要素のみを返します。ここでは range(len(listA)) をデータ側に渡すことで、True値を持つインデックスだけを効率的に抽出できます。

コード例

from itertools import compress
listA = [True, False, 1, False, 0, True]
# 元のリストを表示
print("元のリスト :\n ", listA)
# compress() を使用
res = list(compress(range(len(listA)), listA))
# 結果を表示
print("True値を持つインデックス :\n ", res)

実行結果

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

元のリスト :
[True, False, 1, False, 0, True]
True値を持つインデックス :
[0, 2, 5]

まとめ

どちらの方法も同じ結果になりますが、enumerate() を使った内包表記は可読性が高く、追加のインポートが不要な点がメリットです。一方、compress() はC言語レベルで実装されているため、大きなリストを扱う場合に高速に動作する傾向があります。用途やコードベースのスタイルに合わせて使い分けるとよいでしょう。