Pandas Dataframe 输出两位小数

Question

我有以下代码，并希望将输出（坡度，截距，R_Value，p_value，str_err）折叠成2个小数点。我该如何在此代码中进行？

for x_col in x_vars:
    for y_col in y_vars:
        
        result_dict = {}
        
        temp_df = df[[x_col, y_col]]
        temp_df = temp_df.dropna(how='any')
        
        print(x_col)
        x = temp_df[x_col]
        y = temp_df[y_col]
        
        slope, intercept, r_value, p_value, std_err = stats.linregress(x,y)
        result_dict['x_col'] = x_col
        result_dict['y_col'] = y_col
        
        result_dict["slope"] =slope
        result_dict["intercept"] =  intercept
        result_dict["r-value"] = r_value,
        result_dict["p-value"] = p_value, 
        result_dict["std"] = std_err
        
        result_list.append(result_dict)

Answer 1

您可以使用 round() 直接对这些值进行四舍五入。

for x_col in x_vars:
    for y_col in y_vars:
        
        result_dict = {}
        
        temp_df = df[[x_col, y_col]]
        temp_df = temp_df.dropna(how='any')
        
        print(x_col)
        x = temp_df[x_col]
        y = temp_df[y_col]
        
        slope, intercept, r_value, p_value, std_err = stats.linregress(x,y)
        result_dict['x_col'] = x_col
        result_dict['y_col'] = y_col
        
        result_dict["slope"] = round(slope, 2)
        result_dict["intercept"] = round(intercept, 2)
        result_dict["r-value"] = round(r_value, 2)
        result_dict["p-value"] = round(p_value, 2) 
        result_dict["std"] = round(std_err, 2)
        
        result_list.append(result_dict)

或者，如果要将结果列表转换为数据帧，则可以在列或整个数据帧上使用 pandas.round() 来舍入值（请参阅此处：https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.round.html）

df.round(2)

这个将四舍五入数据框中的值精确到小数点后两位。如果要排除列，请使用 loc[]。

df.loc[:, ['slope', 'intercept', 'r-value', 'p-value', 'std']] = \
    df[['slope', 'intercept', 'r-value', 'p-value', 'std']].round(2)

Answer 2

您可以将函数round（变量，number_of_decimal_places）用于圆形变量。

在您的用例中，您可以使用以下语法来绕过字典值：

#initialize empty dictionary for the rounded values
res_dict = dict()

for key in result_dict:
    if 'col' not in key:
        res_dict[key] = round(result_dict[key], 2)
    else:
        res_dict[key] = result_dict[key]

result_list.append(res_dict)