我有一个 Pandas 数据框。我想保留数字列和特定的列名。我该怎么做?
In [11]: df = pd.DataFrame([[68, 'DC', 'three', 'Sunny', 'Chance of Rain']], columns=['A', 'B', 'C', 'D', 'E'])
In [12]: df.select_dtypes(include=['number'])
Out[12]:
A
0 68
我想创建以下数据框。
A D E
0 68 Sunny Chance of Rain
请您参考如下方法:
我会联合列标题,然后在数据帧上使用列过滤。
df[df.select_dtypes(include=['number']).columns.union(['D', 'E'])]
输出:
A D E
0 68 Sunny Chance of Rain
