python - 按dtype和列名选择Pandas Columns

我有一个熊猫数据框。我想保留数字列和特定的列名称。我怎么做？

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

最佳答案

我会先union列标题，然后在数据框上使用列过滤。

df[df.select_dtypes(include=['number']).columns.union(['D', 'E'])]

输出：

    A      D               E
0  68  Sunny  Chance of Rain

关于python - 按dtype和列名选择Pandas Columns，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/57577331/