Skip to main content
 首页 » 编程设计

python之按 dtype 和列名选择 Pandas 列

2025年05月04日16mate10pro

我有一个 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