比较子集和超集CSV的列,并在Pandas

时间:2018-01-14 12:37:14

标签: python pandas csv

假设我有一个名为super的CSV文件,其中包含列 -

NumVal

1,10

2,25

3,85

4,87

我还有另一个CSV文件sub,其中有一列 -

Num

2

4

现在我想要的是表格的输出 -

NumVal

2,25

4,87

我怎样才能在熊猫中实现这一目标?

1 个答案:

答案 0 :(得分:0)

很难用这种详细程度来猜测你想要实现的目标。但这是第一次试验:这是你在找什么?

import pandas as pd
super = pd.read_csv("super.csv")
sub = pd.read_csv("sub.csv")
# Option 1 - using [merge][1]
result = pd.merge(super, sub, on="Num", how="inner")
# Option 2 - using the [isin][1] method
result = super.loc[super["Num"].isin(sub["Num"].unique())]