如何在 pandas 中向下转换数字列？

问题描述

   A        B    C         D
0  1  1000000  1.1  1.111111
1  2 -1000000  2.1  2.111111

>>> df.dtypes
A      int64
B      int64
C    float64
D    float64

预期结果：

>>> df.dtypes
A       int8
B      int32
C    float32
D    float32
dtype: object

推荐答案

您可以在to_numeric中使用参数downcast，通过DataFrame.select_dtypes选择整型和浮点型列，它使用的是 pandas 0.19+，就像提到的@Anurag，谢谢：

fcols = df.select_dtypes('float').columns
icols = df.select_dtypes('integer').columns

df[fcols] = df[fcols].apply(pd.to_numeric, downcast='float')
df[icols] = df[icols].apply(pd.to_numeric, downcast='integer')

print (df.dtypes)
A       int8
B      int32
C    float32
D    float32
dtype: object