从另一个数据帧的行值中提取一个数据帧中的特定行值

时间:2019-04-25 19:10:38

标签: pandas dataframe data-extraction

我有一个像这样的数据帧(df1):

           X        Y
1          200.0    50            
2          200.1    57    
3          200.2    69
4          200.3    77
5          200.5    84
6          200.6    93

,我还有另一个数据框(df2),如下所示:

           X
1          200.0                
2          200.5    

我想将与df2的X值匹配的df1的Y值提取到看起来像这样的df2中:

           X        Y
1          200.0    50                 
2          200.5    84

如何使用pandas和numpy解决此问题?不幸的是,我是python的新手,我不知道。

谢谢。

最好的问候, DocSnyda

1 个答案:

答案 0 :(得分:0)

pd.merge()是我们提出“在另一个df中查找内容”的要求时首先想到的事情,但是df.loc[]本身也具有“查找内容”的含义。 / p>

"""set the df1 and df2 up """
import pandas as pd
import numpy as np

s ="""20000
20000
20000
200.4
200.5
200.6"""  
df1 = pd.DataFrame(s.split('\n'), columns=['X'])

df1['Y'] = """50
57
69
77
84
93""".split('\n')

df2 = df1.iloc[[0, 5], :]
df2 = df2.drop(columns=['Y'])
print(df1)
print(df2)



""" the anwser to your question here: """
print(
    df1.loc[df1.X.isin(df2.X), :].drop_duplicates(subset='X')
)