返回在另一列中找到其子字符串的行的列表

2024-04-26 11:53:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我有以下数据集

enter image description here

我想以这样一个专栏结束

enter image description here

理想情况下,我希望将列转换为相同的大小写,并按空格拆分字符串,并返回包含在另一列上找到的子字符串的行


Tags: 数据字符串情况空格理想专栏
1条回答
网友
1楼 · 发布于 2024-04-26 11:53:46

检查^{}^{}拆分的第一列的值和第二列的扁平拆分值,并按^{}获得每行至少一个True值,通过筛选第一列传递到^{},如有必要,按^{}创建一列Dataframe

df = pd.DataFrame({'column_a':['ga lt','ka','ku','na ma',np.nan, np.nan],
                   'column_b':['se','ga','ma po','na','ka ch', 'wa wo']})


vals = [y for x in df['column_b'] for y in x.split()]
mask = df['column_a'].str.split(expand=True).isin(vals).any(axis=1)

df = df.loc[mask, 'column_a'].to_frame('column_a_in_column_b')
print (df)
  column_a_in_column_b
0                ga lt
1                   ka
3                na ma

相关问题 更多 >