举例,我有个DataFrame有很多列的数据,现在我想根据手中不属于这个df的其他特定列的数据去匹配这个df得到我想要的数据。
import pandas as pd
df = pd.DataFrame({
'name': ['Jack', 'Lucy', 'Mike', 'Lee'],
'age': ['23', '25', '18', '20'],
'gender': ['male', 'female', 'male', 'male']
})
aaa = ['18', '23', '5', '6']
# 将列表转换为DataFrame
list_df = pd.DataFrame(aaa, columns=['age'])
# 合并DataFrame和列表的DataFrame
merged_df = pd.merge(df, list_df, on='age')
# 显示结果
print(merged_df)
我想用列aaa去匹配df['age'],只保留列aaa和df['age']数据相同的
思路:把列aaa转换为dataframe类型,然后两个df通过merge函数合并