我想获得pandas数据帧中列的相对索引

时间:2015-05-15 18:58:36

标签: python indexing pandas

我想为一只股票的5天回报创建一个新专栏,让我们说。我正在使用pandas数据帧。我使用rolling_mean函数计算了移动平均线,但我不确定如何像电子表格(B6-B1)中那样引用线条。有谁知道我怎么做这个索引参考和减法?

示例数据框:

<div id="background">
  <div id="form-wrapper">
    <div id="form1">
      <form>some form code here</form>
    </div><div id="form2">
      <form>some form code here</form>
    </div>
  </div>
</div>

1 个答案:

答案 0 :(得分:3)

Are you wanting this:

In [10]:

df['5-day-return'] = (df['price'] - df['price'].shift(5)).fillna(0)
df
Out[10]:
   day  price  5-day-return
0    1     10             0
1    2     11             0
2    3     15             0
3    4     14             0
4    5     12             0
5    6     18             8
6    7     20             9
7    8     19             4
8    9     21             7
9   10     22            10

shift返回特定偏移量的行,我们用它来减去这个从当前行。 fillna填写第一次有效计算之前的NaN值。