I have two 3D numpy arrays. I want to compare b with a and then keep only elements in b that exist in a for axis=0.
a= np.asarray([[[-0.5, -0.5, 0.5], [-