按dtype和列名选择Pandas Columns

Question

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

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

Answer 1

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

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

输出：

    A      D               E
0  68  Sunny  Chance of Rain

Answer 2

首先使用select_dtypes，然后使用concat

pd.concat([df.select_dtypes('number'),df[['C','D']]],axis=1)
Out[369]: 
    A      C      D
0  68  three  Sunny