toc*_*eto 11 python dataframe pandas
我有2个数据帧.
Df1 = pd.DataFrame({'name': ['Marc', 'Jake', 'Sam', 'Brad']
Df2 = pd.DataFrame({'IDs': ['Jake', 'John', 'Marc', 'Tony', 'Bob']
Run Code Online (Sandbox Code Playgroud)
我想循环遍历每一行Df1['name']并检查每个名字是否在某处Df2['IDs'].
如果名称在那里,结果应返回1,如果不是这样,则返回0:
Marc 1
Jake 1
Sam 0
Brad 0
Run Code Online (Sandbox Code Playgroud)
谢谢.
piR*_*red 18
使用 isin
Df1.name.isin(Df2.IDs).astype(int)
0 1
1 1
2 0
3 0
Name: name, dtype: int32
Run Code Online (Sandbox Code Playgroud)
在数据框中显示结果
Df1.assign(InDf2=Df1.name.isin(Df2.IDs).astype(int))
name InDf2
0 Marc 1
1 Jake 1
2 Sam 0
3 Brad 0
Run Code Online (Sandbox Code Playgroud)
在一个Series对象中
pd.Series(Df1.name.isin(Df2.IDs).values.astype(int), Df1.name.values)
Marc 1
Jake 1
Sam 0
Brad 0
dtype: int32
Run Code Online (Sandbox Code Playgroud)
这应该这样做:
Df1 = Df1.assign(result=Df1['name'].isin(Df2['IDs']).astype(int))
Run Code Online (Sandbox Code Playgroud)
通过使用 merge
s=Df1.merge(Df2,left_on='name',right_on='IDs',how='left')
s.IDs=s.IDs.notnull().astype(int)
s
Out[68]:
name IDs
0 Marc 1
1 Jake 1
2 Sam 0
3 Brad 0
Run Code Online (Sandbox Code Playgroud)
这是一种方式。转换为 O(1) 查找设置并用于astype(int)将布尔值表示为整数。
values = set(Df2['IDs'])
Df1['Match'] = Df1['name'].isin(values).astype(int)
Run Code Online (Sandbox Code Playgroud)