python - 如何在列上添加标题

Question

如何将列添加到数据框 xFlattened ？就我而言，xFlattened.columns 没有帮助我。

def corrFilter(x, bound):
    xCorr = x.corr()
    xFiltered = xCorr[((xCorr >= bound) | (xCorr <= -bound)) & (xCorr !=1.000)]
    xFlattened = xFiltered.unstack().sort_values(ascending = False).drop_duplicates()
    xFlattened.columns = ["first_value", "second_value", "corr"]
    return xFlattened

d = {'review_scores_accuracy': [1.1, 2.0, 4.5, 5.0, 4.9, 4.8, 4.7], 
     'review_scores_rating': [1.1, 2.0, 4.6, 3.9, 4.2, 4.5, 4.2]}
df = pd.DataFrame(data=d)

df = corrFilter(df, .7)
df

我想要什么

             first_value              second_value      percentage
0 review_scores_accuracy      review_scores_rating        0.968217
1                           review_scores_accuracy             NaN

Answer 1

您需要在函数中添加 reset_index 才能将 Series 转换为 DataFrame:

xFlattened = xFiltered.unstack().sort_values(ascending = False).drop_duplicates().reset_index()

输出:

              first_value            second_value      corr
0  review_scores_accuracy    review_scores_rating  0.968217
1  review_scores_accuracy  review_scores_accuracy       NaN