Pythonでリストから指定したインデックスの複数要素を削除する方法

Pythonのリストから単一の要素を削除するのは、del文とインデックスを組み合わせれば簡単に行えます。しかし、複数のインデックスに該当する要素をまとめて削除したいケースでは、少し工夫が必要です。本記事では、削除対象となるインデックスのリストを指定して、元のリストから該当する要素だけを取り除く方法を2つ紹介します。

方法1：sorted()とdelを組み合わせる

このアプローチでは、まず削除したい位置（インデックス）を格納したリストを作成します。その後、降順にソートしてから後ろの要素から順に削除することで、削除処理中にインデックスがずれる問題を回避し、元のリストの構造を保ったまま安全に要素を削除できます。最後に、del文を使って元のリストから該当するインデックスの要素を削除します。

サンプルコード

Alist = [11, 6, 8, 3, 2]

# 削除対象のインデックスリスト
idx_list = [1, 3, 0]

# 元のリストを表示
print("Given list is : ", Alist)

# インデックスリストを表示
print("The indices list is : ", idx_list)

# 降順ソートしてdelで削除
for i in sorted(idx_list, reverse=True):
    del Alist[i]

# 結果を表示
print("List after deleted elements : ", Alist)

実行結果

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

Given list is : [11, 6, 8, 3, 2]
The indices list is : [1, 3, 0]
List after deleted elements : [8, 2]

idx_listは降順ソートにより [3, 1, 0] の順で処理されるため、大きなインデックスから先に削除されます。これにより、削除のたびにリストが縮んでも、まだ処理していない小さいインデックスの位置がずれることはありません。結果として、インデックス0・1・3にある要素だけが正しく削除され、[8, 2] が出力されます。

方法2：enumerate()とnot inを使う

もうひとつの方法として、enumerate()関数と内包表記の中で not in を使う手法があります。各要素とともにインデックスを取得し、そのインデックスが削除対象リストに含まれていない場合のみ新しいリストに残すことで、同じ結果を得られます。スライス代入 Alist[:] を使うことで、元のリストオブジェクト自体を書き換えている点もポイントです。

サンプルコード

Alist = [11, 6, 8, 3, 2]

# 削除対象のインデックスリスト
idx_list = [1, 3, 0]

# 元のリストを表示
print("Given list is : ", Alist)

# インデックスリストを表示
print("The indices list is : ", idx_list)

# スライス代入とnot inで除外
Alist[:] = [ j for i, j in enumerate(Alist)
             if i not in idx_list ]

# 結果を表示
print("List after deleted elements : ", Alist)

実行結果

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

Given list is : [11, 6, 8, 3, 2]
The indices list is : [1, 3, 0]
List after deleted elements : [8, 2]

まとめ

どちらの方法も、指定した複数のインデックスに対応する要素を一度に削除できる便利なテクニックです。sorted(reverse=True)とdelの組み合わせはシンプルで直感的であり、一方、enumerate()と内包表記を使う方法はPythonらしい宣言的なスタイルで記述できます。用途やコードの可読性に応じて使い分けるとよいでしょう。