2 x 2 x 3のndarrayを要素比較した上で2 x 2 x 1のndarrayに変換する方法をご教授頂けませんでしょうか

Question

###前提・実現したいこと

2 x 2 x 3のndarrayを要素比較して2 x 2 x 1のndarrayに変換したいです。

###該当のソースコード

In [1]: xs = np.array([[[1, 0, 0], [2, 0, 0]], [[1, 0, 0], [2, 0, 0]]], dtype=np.uint8)

In [2]: xs
Out[2]: 
array([[[1, 0, 0],
        [2, 0, 0]],

       [[1, 0, 0],
        [2, 0, 0]]], dtype=uint8)

上記のndarrayの各要素とys = np.array([1, 0, 0], dtype=np.uint8)を比較し、下記のような変換を行いたいと思っています。

In [3]: result
Out[3]: 
array([[[True],
        [False]],

       [[True],
        [False]]], dtype=bool)

こちらについて実現可能であれば、その方法をご教授頂けませんでしょうか？
宜しくお願い致します。

###補足情報(言語/FW/ツール等のバージョンなど)

• Python 3.5.2 :: Anaconda 2.5.0 (x86_64)
• numpy 1.12.0

Answer 1

これでどうでしょう

>>> (xs == ys).all(2)
array([[ True, False],
       [ True, False]], dtype=bool)

>>> (xs == ys).all(2, keepdims=True)
array([[[ True],
        [False]],

       [[ True],
        [False]]], dtype=bool)

Comments

[9zZBqEam]

ご回答頂きまして大変ありがとうございました。
NumPyのAPIで実装されており処理速度も十分かと思われますので、こちらベストアンサーとさせて頂きます。

Answer 2

あまり綺麗なコードになりませんでしたが、一応できました。

np.array([y.tolist() == [1, 0, 0] for x in xs for y in x]).reshape((2, 2, 1))
array([[[ True],
        [False]],

       [[ True],
        [False]]], dtype=bool)

Comments

[9zZBqEam]

ご回答頂きまして大変ありがとうございました。