您可以使用
NumPy'sin1d-
df.index[np.in1d(df['Col4'],target_array)]
说明
1)创建一个
1D与每一行相对应的掩码,以告诉我们
col4's元素和中的任何元素之间是否存在匹配项
target_array:
mask = np.in1d(df['Col4'],target_array)
2)使用掩码从数据框中选择有效索引作为最终输出:
out = df.index[np.in1d(df['Col4'],target_array)]
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)