python - pandas 中的回归列

Question

假设我有一个 pandas 数据框 df ，带有一些简单的索引，例如0,1,2,...，只有一列包含数值数据的'Values'。我想添加一个新列 'Trend' ，这样 df['Trend'][n] 是基于原始值的 5 个连续值的线性回归系数值列:'df['Values'][n-5] ... 'df['Values'][n-1]。前 5 个值当然是未定义的，所以假设我们将它们设置为 NaN。有没有简单的方法可以做到这一点？

Answer 1

您是否尝试过检查索引并根据您在DataFrame中的位置执行不同的操作？

for index, row in df.iterrows():
    if index<5:
        df['Trend'][index] = 'NaN'
    else:
        df['Trend'][index] = df['Values'][index-5]...