解决pandas 四舍五入异常问题

杨白告

于 2024-05-22 12:57:56 发布

阅读量427

点赞数 5

分类专栏： python 文章标签： pandas

本文链接：https://blog.csdn.net/weixin_48984179/article/details/139117256

版权

python 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

import pandas as pd
import decimal

def count_employees(employees: pd.DataFrame) -> pd.DataFrame:
    D = decimal.Decimal
    decimal.getcontext().rounding = "ROUND_HALF_UP" # 设置舍入方式为四舍五入

    emp_table = employees[~employees["reports_to"].isnull()]

    re_table = employees[["employee_id","name"]].merge(emp_table[["reports_to","age"]],left_on="employee_id",right_on="reports_to",how="left")

    re_table = re_table.dropna(subset=["reports_to"])

    return re_table.groupby(["employee_id","name"]).agg(
        reports_count=("reports_to","count"),
        average_age = ("age",lambda x:D(x.mean()).quantize(D("0")))).reset_index() # 对应pandas四舍五入问题,主要看这个