numpy.any, allを使って1行で書く方法を探しています。

Question

arr=
[[1, 0],
 [2, 0],
 [2, 1],
 [0, 3],
 [3, 1],
 [0, 2]]

lst=
[1, 2]

のようなN×2のnumpy行列arrと、1次元行列lstがあるときに、
arrのすべての行のうち、第1要素(右側)がlstに含まれるような行だけを抽出したいのですが、どうすればよいでしょうか？

たとえばこの場合は、

[[2, 1],
 [3, 1],
 [0, 2]]

となるようにしたいです。
関数として書くと、

def my_func(arr, lst):
    result = []
    for i in arr:
        if np.any(lst == i[1]):
            result.append(i)
    return np.asarray(result)

となります。
numpyの機能を使って1行で書くことができればうれしいのですが、どうすればよいでしょうか？

Answer 1

こんなふうに書けるのかな、と。

arr[np.isin(arr[:,1], lst),:]

Answer 2

numpy.any, allは使っていませんが、1行で書いてみました。

arr[[i in lst for i in arr[:,1]]]