我有一个像这样的Numy数组:
source = np.array([[[0,0,0],[0,0,1],[0,1,0],[1,0,0],[1,0,1],[1,1,0],[1,1,1]]])
我正在try 将它与另一个数组进行比较,后者的Axis2更短,Axis3中有重复项:
values = np.array([[[0,1,0],[1,0,0],[1,1,1],[1,1,1],[0,1,0]]])
我的目标是有一个布尔数组,只要最长的:
[False, False,True,True,False,False,True]
我试过这些命令:
np.isin(source,values).all(axis=2)
但它显示了一个由七个True组成的array.像numpy.in1d()这样的函数似乎是一个很好的 Select ,但我没有实现将其改编为适用于3Darray.