根据列名值对 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)]