python 移动平均 连续三个空值 移动平均python代码

# http://localhost:8888/lab
# jupyter lab
import tushare as ts
import numpy as np
# 导入画图工具matplotlib
import matplotlib.pyplot as plt
import pandas as pd

# 初始化pro接口
pro = ts.pro_api('')

print("获取股票数据.")
zgpa = pro.daily(ts_code='000006.SZ', start_date='20190101', end_date='20190301')

zgpa_signal = pd.DataFrame(index=zgpa.index)

# 这里使用10日均线
period = 10
# 设置一个空列表，用来存储每10天的价格
avg_10 = []
# 再设置一个空列表，用来存储每10天价格的均值
avg_value = []
# 设置一个循环
for price in zgpa['pre_close']:
    # 把每天的价格传入到avg_10列表
    avg_10.append(price)
    print('++++++++++')
    print(avg_10)
    print(np.mean(avg_10))
    # 当列表中存储的数值多于10个时
    if len(avg_10) > period:
        # 就把前面传入的价格数据删掉，确保列表中只有10天的数据
        del avg_10[0]
    # 将10天数据的均值传入到avg_value列表中
    avg_value.append(np.mean(avg_10))
# 把计算好的10日均价写到股票价格数据表中
zgpa = zgpa.assign(avg_10=pd.Series(avg_value, index=zgpa.index))
# 检查一下是否添加成功
print(zgpa)

# 设置图像尺寸为10*6
plt.figure(figsize=(10, 6))
# 绘制股价的变化
plt.plot(zgpa['pre_close'], lw=2, c='k')
# 绘制10日均线
plt.plot(zgpa['avg_10'], '--', lw=2, c='b')
# 添加图注和网格
plt.legend()
plt.grid()
# 将图像进行显示
plt.show()