2017-06-16 63 views
1

我有一個關於數據幀合併的基本問題。合併兩個數據框後,是否有一種方法可以在結果中只選取幾列。Python Pandas Dataframe合併並挑選只有幾列

以從文檔

https://pandas.pydata.org/pandas-docs/stable/merging.html#

left = pd.DataFrame({'key1': ['K0', 'K0', 'K1', 'K2'], 
        'key2': ['K0', 'K1', 'K0', 'K1'], 
        'A': ['A0', 'A1', 'A2', 'A3'], 
        'B': ['B0', 'B1', 'B2', 'B3']}) 


right = pd.DataFrame({'key1': ['K0', 'K1', 'K1', 'K2'], 
          'key2': ['K0', 'K0', 'K0', 'K0'], 
          'C': ['C0', 'C1', 'C2', 'C3'], 
          'D': ['D0', 'D1', 'D2', 'D3']}) 



result = pd.merge(left, right, on=['key1', 'key2']) 

結果的一個例子當屬:

A B key1 key2 C D 
0 A0 B0 K0 K0 C0 D0 
1 A2 B2 K1 K0 C1 D1 
2 A2 B2 K1 K0 C2 D2 
None 

有沒有一種方法可以讓我從 '右' 數據框中只選擇了列 'C'?例如,我想我的結果是這樣的:

A B key1 key2 C 
0 A0 B0 K0 K0 C0 
1 A2 B2 K1 K0 C1 
2 A2 B2 K1 K0 C2 
None 

回答

1
result = pd.merge(left, right[['key1','key2','C']], on=['key1', 'key2']) 

OR

right.merge(left, on=['key1','key2'])[['A','B','C','key1','key2']] 
+0

大。我看到你過濾了正確的數據框的列。 – ProgSky