in関数を使った判定について

Question

例えば、

a=[[1,1],[2,2]]
if [1,2] in a:
  print('ok') 

としたときにはokは出力されないのですが、
（aの中に[1,2]の要素があるかどうかを判定しているため？）

a=numpy.array([[1,1],[2,2]])
if [1,2] in a:
  print('ok')

とすると、okが出力されてしまいます。
（aの中に1または2があるかどうかを判定？）

aをリストではなくnumpy配列としたときでも、上のコードと同じように指定した行があるかどうかで判定することはできないのでしょうか？

Answer 1

https://github.com/numpy/numpy/blob/491f41acd2c6a0e7886662a2448b3d245ae432a9/numpy/core/src/multiarray/sequence.c#L29

equivalent to (self == el).any()

だそう。

==の動作は

>>> a=numpy.array([[1,2],[3,4]])  # わかりやすいように質問とは異なります

>>> a
array([[1, 2],
       [3, 4]])

>>> a == [1,2]
array([[ True,  True],
       [False, False]])

>>> a == [1,1]
array([[ True, False],
       [False, False]])

>>> a == [1,4]
array([[ True, False],
       [False,  True]])

>>> a == [3,2]
array([[False,  True],
       [ True, False]])

こうなので、ここに一つでもTrueがあったらin演算子はTrueになります。

---

欲しい動作は

1. ==を適用した結果に
2. 横に見た時にすべてTrueなものが
3. 1つでもあれば

という条件になるので、

>>> (a == [1,2]).all(axis=1)  # 2. 横に見た時にすべてTrueか
array([ True, False])


>>> (a == [1,2]).all(axis=1).any()  # 3. ↑の結果に一つでもTrueがあるか
True

かと。

a = numpy.array([[1,1],[2,2]])
if (a == [1,2]).all(axis=1).any():
  print('ok') 

出力なし

---

（問題はもっと明快な書き方がないのかということ）

Comments

[rakkyo11]

そのような方法もあるのですね、勉強になりました。回答ありがとうございました。

Answer 2

やはり、リストやタプルなどにするしか無いような。

a=numpy.array([[1,1],[2,2]])
aa=[list(_) for _ in a]
if [1,2] in aa:
    print('ok')

Comments

[rakkyo11]

numpy配列のままするのは難しんですね...回答ありがとうございます。

[x98000]

似たような事例がありました。小難しくなりますがデータ量が多い場合は有効かもしれません。
https://stackoverflow.com/questions/8317022/get-intersecting-rows-across-two-2d-numpy-arrays/