根據列名值對 pandas 中的 dataframe 進行子集化

Question

我在 pandas 中有一個 dataframe 需要拆分。 它比這大得多，但這里有一個例子：

ID A  B

a   0  0

b   1  1 

c   2  2

我有一個列表： keep_list = ['ID','A']和另一個列表： recode_list = ['ID','B']

我希望通過列標題將 dataframe 拆分為兩個數據幀：一個 dataframe 具有列標題與 keep_list 匹配的列和值，另一個具有與 recode_alleles 列表匹配的列標題和數據。 到目前為止，我嘗試過的每個代碼都沒有工作，因為它試圖將值與列表進行比較，而不是列名。

非常感謝您的幫助！

Answer 1

假設您的 DataFrame 的名稱是df ：

你可以簡單地做df[keep_list]和df[recode_list]來得到你想要的。

Answer 2

您可以通過index.intersection執行此操作：

df1 = df[df.columns.intersection(keep_list)]
df2 = df[df.columns.intersection(recode_list)]