def where_list(array,values):
mask = nd.sum(np.ones_like(values),values,index=array)
return np.where(mask>0)
微信
公众号
扫码关注公众号
微信
公众号