从两个集合的交集中选择数据帧的子集

Question

我从一个数据帧中得到了两组列标题。一个集合是另一个集合的子集。你知道吗

import pandas as pd
d = {'feature1':[1,2,3], 'feature2':[3,4,5], 'feature3':[6,7,8]}
df = pd.DataFrame(data=d)

现在我有两套：

set_1 = {'feature1','feature2','feature3'}
set_2 = {'feature1','feature3'}

如何在我的（新）数据帧中仅选择子集中的列？你知道吗

Answer 1

集合是不可散列的，因此您需要将它们转换为一个列表，例如。你知道吗

然后可以像这样选择数据帧

df[list(set_1)]

它回来了

    feature1    feature3    feature2
0   1   6   3
1   2   7   4
2   3   8   5

或者

df[list(set_2)]

它回来了

    feature1    feature3
0   1   6
1   2   7
2   3   8