Pythonで複雑なデータ型（リスト・タプルなど）を含む行を抽出する方法

Pythonでは、ネストされたデータ構造の中から、リストやタプル、辞書、セットといった複雑なデータ型を含む行だけを抜き出したいケースがあります。このような場合、組み込み関数の isinstance() とリスト内包表記を組み合わせることで、シンプルかつ効率的に処理できます。

サンプルコード

以下は、実際の動作を示した例です。

my_list = [[13, 1, 35], [23, [44, 54], 85], [66], [75, (81, 2), 29, 7]]

my_result = [row for row in my_list if any(isinstance(element, list) or isinstance(element, tuple) or isinstance(element, dict) or isinstance(element, set) for element in row)]

print("元のリスト :")
print(my_list)

print("抽出結果のリスト :")
print(my_result)

実行結果

元のリスト :
[[13, 1, 35], [23, [44, 54], 85], [66], [75, (81, 2), 29, 7]]
抽出結果のリスト :
[[23, [44, 54], 85], [75, (81, 2), 29, 7]]

コードの解説

• まず、複数のリストを要素として持つリスト（リストのリスト）を定義し、その内容をコンソールに表示します。
• 次に、リスト内包表記を使って各行を順番に走査し、isinstance() メソッドで各要素がリスト型かどうかを判定します。この例では、タプル・辞書・セットについても同様にチェックしています。
• any() 関数により、行の中に1つでも複雑なデータ型が含まれていれば条件がTrueとなり、その行が抽出対象となります。
• 抽出された結果は変数に代入され、最後にコンソールへ出力されます。

より簡潔に書くテクニック

isinstance() の第2引数にはタプルを渡せるため、複数の型をまとめて判定できます。これを利用すると、先ほどのコードは以下のようにすっきりと書き換えられます。

my_result = [row for row in my_list if any(isinstance(element, (list, tuple, dict, set)) for element in row)]

この書き方なら、判定したい型を追加する場合もタプルに要素を加えるだけで済むため、保守性の高いコードになります。