图片数据的3D numpy数组处理

Question

我有两个numpy数组。一种是例如用于图像数据，第二种是用于相应的图像标签。例如，

import numpy as np
 ## image data
a = np.array([[[1,5,9],[7,5,9],[2,5,5]], [[3,8,5],[7,15,10],[1,0,9]], [[8,15,0],[1,1,15],[5,5,0]], [[4,5,10],[10,10,1],[4,5,10]]])
## corresponding label
b = np.array([0,1,2,1])

我正在寻找一种获取新图像数据的方法，该图像数据仅包含图像标签0和1的值。对于我来说，我希望获得新的结果，

a_new = np.array([[[1,5,9],[7,5,9],[2,5,5]], [[3,8,5],[7,15,10],[1,0,9]], [[4,5,10],[10,10,1],[4,5,10]]])
b_new = np.array([0,1,1])

我们可以在numpy中执行此操作而不使用任何for循环吗？谢谢有人帮助我。