我有以下数据帧
df = pd.DataFrame({'Id':['1','2','3'],'List_Origin':[['A','B'],['B','C'],['A','B']]})
我怎么能只获取ID,它只包含某个List\u原点,例如"a"、"B".如果解决方案能够避免循环,我们将不胜感激
想要的最终结果
end_df = pd.DataFrame({'Id':['1','3'],'List_Origin':[['A','B'],['A','B']]})
我有以下数据帧
df = pd.DataFrame({'Id':['1','2','3'],'List_Origin':[['A','B'],['B','C'],['A','B']]})
我怎么能只获取ID,它只包含某个List\u原点,例如"a"、"B".如果解决方案能够避免循环,我们将不胜感激
想要的最终结果
end_df = pd.DataFrame({'Id':['1','3'],'List_Origin':[['A','B'],['A','B']]})
您可以使用apply
并进行如下判断:
>>> df[df['List_Origin'].apply(lambda x: x==['A', 'B'] or x==['A,B'])]
Id List_Origin
0 1 [A,B]
2 3 [A, B]