我有两个数组：3D数组ndarray(N,M,3)和1D数组ndarray(3)。如何通过比较3D数组的最后一个轴和1D数组来创建掩码N×M？
现在，我可以这样做：

mask = (A[:,:,0] == B[0]) & (A[:,:,1] == B[1]) & (A[:,:,2] == B[2])

字符串
但必须有一个更短的解决方案。