Numpy：比较另一个数组中的数组元素

Question

我有一个numpy数组

X = [[1,2], [3,4], [5,6], [1,2], [5,6]]

我想要一个numpy数组Y = [1, 2, 3, 1, 3]，其中[1,2]替换为1，[3,4]替换为2，依此类推。这是一个非常大的（想想数百万）X。

直觉是{​​{1}}。但这确实无效。

Answer 1

  

直觉是{​​{1}}。但这不起作用。

以下是如何使其发挥作用：

Y[X == [1,2]] = 1

处理所有可能的值：

Y = np.empty(len(X), dtype=np.int)
Y[np.all(X == [1, 2], 1)] = 1