python - 按 dtype 选择 Pandas 列

Question

我想知道 Pandas DataFrames 中是否有一种优雅且简写的方式来按数据类型 (dtype) 选择列。即从 DataFrame 中仅选择 int64 列。

详细地说，类似于

df.select_columns(dtype=float64)

提前感谢您的帮助

Answer 1

从 0.14.1 开始，有一个 select_dtypes方法，这样你就可以更优雅/更一般地做到这一点。

In [11]: df = pd.DataFrame([[1, 2.2, 'three']], columns=['A', 'B', 'C'])

In [12]: df.select_dtypes(include=['int'])
Out[12]:
   A
0  1

To select all numeric types use the numpy dtype numpy.number

In [13]: df.select_dtypes(include=[np.number])
Out[13]:
   A    B
0  1  2.2

In [14]: df.select_dtypes(exclude=[object])
Out[14]:
   A    B
0  1  2.2