python 2次元配列内の検索の簡易な方法

Question

test = [["a", 102], ["b", 2022], ["c", 23302], ["d", 4502], ["e", 5022], 
["f", 1602],・・・・・・・・・]

こんな感じの2次元配列がある時
リスト内のリストの1番目(１０２，２０２２，２３３０２，４５０２等)
のみを対象に値の有無を確認するにはどうしたらいいでしょうか？

これが一次元配列でしたら

test = [102, 2022, 23302, 4502, 5022, 1602]
print(5000 not in test)

この様に書けますが、2次元配列内のリスト内の〇番目のみを対象にして値の有無を確認する
命令があれば教えてください

※
for文等を使って一つ一つ取り出して確認する方法は時間がかかりすぎるのとコードが増えて
見辛くなるので
not in の様な短くて済むコードがあればそちらの方をお願いします

Answer 1

自分だったら一次元配列にするかなあ

test = [["a", 102], ["b", 2022], ["c", 23302], ["d", 4502], ["e", 5022], ["f", 1602]]
print(5000 not in [x[1] for x in test])

Comments

[shiracamus]

リスト内包表記にするとリストを作ってから処理開始しますが、ジェネレータ式にすれば都度要素を作って条件一致した時点で処理を中断するので速くなることが期待できますよ。

print(5000 not in (x[1] for x in test))

[takasima20]

そうですね。助言ありがとうございます。

Answer 2

numpy を使う場合。

>>> import numpy as np
>>> test = np.array([
      ["a", 102], ["b", 2022], ["c", 23302],
      ["d", 4502], ["e", 5022], ["f", 1602],
    ], dtype=object)

>>> 5000 not in test[:,1]
True
>>> 1602 not in test[:,1]
False

Answer 3

for文等を使って一つ一つ取り出して確認する方法は時間がかかりすぎるのとコードが増えて見辛くなるので

愚直にループするしかないと思いますけど、map, any, all とジェネレータ式を使えば、ひとつ見つかった時点で処理を終了するので多少は速いです。

test = [["a", 102], ["b", 2022], ["c", 23302], ["d", 4502], ["e", 5022], ["f", 1602],]

print(5000 not in map(lambda t: t[1], test))

test = [["a", 102], ["b", 2022], ["c", 23302], ["d", 4502], ["e", 5022], ["f", 1602],]

print(all(t[1] != 5000 for t in test))

test = [["a", 102], ["b", 2022], ["c", 23302], ["d", 4502], ["e", 5022], ["f", 1602],]

print(not any(t[1] == 5000 for t in test))