Python/Pandas Dataframe - 当 0 时将小数精度替换为 int 值，否则四舍五入到 n 位小数

Question

当值为 3.0 时，使用以下构造删除小数精度 (. & 0)；当值为 3.12345 时，四舍五入到小数点后 4 位

import pandas as pd
df1 = pd.DataFrame({'Price':[1.0,2.12345,3.0,4.67892]})
df1["Price"] = df1["Price"].apply(lambda x: round(x,4) if x%1 else int(x))
print(df1)

舍入有效，但转换为 int 无效。

Answer 1

您需要使用dtype=object将该列转换为对象类型:

df1["Price"] = np.array([int(x) if x%1==0 else round(x,4) for x in df1["Price"].values ], dtype=object)

    Price
0       1
1  2.1234
2       3
3  4.6789

如下所示，对象在必要时保存 int 和 float:

[print (type(i)) for i in df1["Price"].values]

Out[1]
<class 'int'>
<class 'numpy.float64'>
<class 'int'>
<class 'numpy.float64'>