从两组交集中选择数据框的子集

Question

我从DataFrame中获得了两组列标题。一组是另一组的子集。

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

集不可散列，因此，例如，您需要将它们转换为列表。

然后您可以选择这样的数据框

TENANT_SCHEMA

返回

os.environ

或

env.py

返回

df[list(set_1)]