替换 numpy 中的子数组

给定一个数组，

>>> n = 2
>>> a = numpy.array([[[1,1,1],[1,2,3],[1,3,4]]]*n)
>>> a
array([[[1, 1, 1],
        [1, 2, 3],
        [1, 3, 4]],

       [[1, 1, 1],
        [1, 2, 3],
        [1, 3, 4]]])

我知道可以像这样简洁地替换其中的值，

>>> a[a==2] = 0
>>> a
array([[[1, 1, 1],
        [1, 0, 3],
        [1, 3, 4]],

       [[1, 1, 1],
        [1, 0, 3],
        [1, 3, 4]]])

是否可以对数组中的整行（最后一个轴）执行相同的操作？我知道a[a==[1,2,3]] = 11将工作并将匹配子数组的所有元素替换为 11，但我想替换不同的子数组。我的直觉告诉我写以下内容，但结果出现错误，

>>> a[a==[1,2,3]] = [11,22,33]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: array is not broadcastable to correct shape

总而言之，我想要得到的是：

array([[[1, 1, 1],
        [11, 22, 33],
        [1, 3, 4]],

       [[1, 1, 1],
        [11, 22, 33],
        [1, 3, 4]]])

...当然，一般来说，n 比 2 大很多，并且其他轴也比 3 大，所以如果不需要，我不想循环它们。

更新：[1,2,3]（或我正在寻找的任何其他内容）并不总是位于索引 1。一个例子：

a = numpy.array([[[1,1,1],[1,2,3],[1,3,4]], [[1,2,3],[1,1,1],[1,3,4]]])

您可以使用更高的性能来实现这一点np.all检查所有列是否都有True进行比较的值，然后使用创建的掩码替换这些值：

mask = np.all(a==[1,2,3], axis=2)
a[mask] = [11, 22, 23]

print(a)
#array([[[ 1,  1,  1],
#        [11, 22, 33],
#        [ 1,  3,  4]],
# 
#       [[ 1,  1,  1],
#        [11, 22, 33],
#        [ 1,  3,  4]]])