Blog Content

pandas的dataframe计算连续日期均线、移动平均线rolling方法

Python 智能投研 统计学-科学计算   2016-12-16 11:23:35

---

计算连续日期的移动均线

import pandas as pd
import numpy as np
 
print(type(np.random.randint(1, 100, size=50)))
 
df = pd.DataFrame([np.random.randint(1, 10, size=4),
                   np.random.randint(1, 10, size=4),
                   np.random.randint(1, 10, size=4),
                   np.random.randint(1, 10, size=4),
                   np.random.randint(1, 10, size=4),
                   np.random.randint(1, 10, size=4)], columns=["first", "second", "third", "four"])
 
print(df)
 
 
# 计算5个窗口位置的平均线【如果查询的是day日期数据,那么这里等价于2日均线】
print(df["first"].rolling(2).mean())
 
print("--------------------")
 
# 计算5个窗口位置的移动平均线【如果查询的是day日期数据,那么这里等价于2日移动均线】
print(df["first"].rolling(2).mean().shift(-1))

---

上一篇：elasticsearch多个索引的数据的联合查询
下一篇：pandas和numpy中nan值的判断