python - Pandas 只更改 dtypes 的 float64 列

Question

我需要更改多列(超过 400)的 dtype，但数据框具有不同类型的 dtype。一些列数据类型是 float64而有些列是 int64或 object :

print my_df.dtypes

输出:

x1                       int64
x2                       int64
x3                       object
x4                       float64
x5                       float64
x6                       float64
x7                       float64
...

x400                     object
x401                     object
x402                     object
...

我需要更改所有 int64至 int8或 int16以及所有 float64至 float32 .我试过下面的代码片段，但没有奏效:

my_df[my_df.dtypes == np.int64].astype(np.int16)
my_df[my_df.dtypes == np.float64].astype(np.float32)

任何帮助表示赞赏。

提前致谢。

Answer 1

好的，我找到了我的方式:)

查找数据类型为 float64 的列

cols = my_df.select_dtypes(include=[np.float64]).columns

然后只更改数据类型 cols的数据帧。

my_df[cols] = my_df[cols].astype(np.float32)