Pythonで行列のN番目の列を判定する方法｜any()関数で特定列の要素存在をチェック

Pythonで行列（リストのリスト）を扱う際、「N番目の列に特定の要素が存在するかどうか」を確認したいケースがあります。そんなときに便利なのが、組み込み関数 any() です。any() を使えば、明示的なループ処理を書かずとも、簡潔なコードで条件を満たす要素の有無を判定できます。

以下に具体的な実装例を示します。

サンプルコード

my_list = [[34, 67, 89], [16, 27, 86], [48, 30, 0]]

print("リストの内容:")
print(my_list)

N = 1
print("N の初期値:")
print(N)
elem = 30

my_result = any(sub[N] == elem for sub in my_list)

print("指定した列に要素は存在するか?")
print(my_result)

実行結果

リストの内容:
[[34, 67, 89], [16, 27, 86], [48, 30, 0]]
N の初期値:
1
指定した列に要素は存在するか?
True

コードの解説

• まず、リストの中にリストを格納した2次元リスト（行列）を定義し、コンソールに表示します。

• 変数 N に、確認したい列のインデックスを設定します。インデックスは0から始まるため、N = 1 は2列目を指します。

• この値をコンソールに表示します。

• 変数 elem に、検索対象となる整数値を代入します。

• any() 関数にジェネレータ式を渡し、各行 sub の sub[N]（N番目の列の要素）が elem と一致するかを順に評価します。一つでも一致すれば True を返します。

• 判定結果を変数 my_result に格納します。

• 最後に、結果をコンソールに出力します。

ポイント

any() は、引数として渡されたイテラブルの要素が一つでも真であれば True を返す組み込み関数です。ジェネレータ式と組み合わせることで、全行を走査しながら「N番目の列に目的の要素が存在するか」を効率的に判定できます。また、条件を満たす要素が見つかった時点で評価を打ち切るため、大規模なデータでも高速に動作します。

なお、逆に「すべての行のN番目の要素が一致すること」を確認したい場合は、同様の書き方で all() 関数を使用すると実現できます。