python 灰色系统预测GM

安装方法

使用pip安装：pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ greytheory

github地址：https://github.com/Kalvar/python-GreyTheory

代码

使用GM(1,1)预测，参考官网代码：

from greytheory import GreyTheory

grey = GreyTheory()
gm11 = grey.gm11

# 依次把真实值加入进来
gm11.add_pattern(223.3, "a1")
gm11.add_pattern(227.3, "a2")
gm11.add_pattern(230.5, "a3")
gm11.add_pattern(238.1, "a4")
gm11.add_pattern(242.9, "a5")
gm11.add_pattern(251.1, "a6")

gm11.period = 2  # 多预测多少个值
gm11.forecast()  # 真实的预测逻辑

gm11.print_forecasted_results()  # 仅仅是打印，所得的数据已经被计算出来了

for value in gm11.analyzed_results:
    print(value.forecast_value)  # 这里依次打印预测值，也可以保存在list中

针对一个pd.Series()，使用如下方法可以快速得到预测结果：

# 传入一个 series，返回预测的列表，gm11.period = 1 时 value_list[-1] 即为向后预测的一个值
def gray_predict(series: pd.Series) -> list:
    grey = GreyTheory()
    gm11 = grey.gm11
    for index, data in enumerate(series):
        gm11.add_pattern(data, str(index))
    gm11.period = 1  # 调整向后预测的数量
    gm11.forecast()
    value_list = []
    for value in gm11.analyzed_results:
        value_list.append(value.forecast_value)
    return value_list