From 0e07e519d97714fe0c5271325e8d6e0308d81117 Mon Sep 17 00:00:00 2001
From: kpt <2817151932@qq.com>
Date: Wed, 3 Jul 2024 22:51:12 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90predict=5Fdemo=E3=80=91=E9=A2=84?=
 =?UTF-8?q?=E6=B5=8B=20demo=EF=BC=8C=E7=94=A8=E4=BA=8E=E6=BC=94=E7=A4=BA?=
 =?UTF-8?q?=E9=A2=84=E6=B5=8B=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 predict_demo.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 predict_demo.py

diff --git a/predict_demo.py b/predict_demo.py
new file mode 100644
index 0000000..5daeda0
--- /dev/null
+++ b/predict_demo.py
@@ -0,0 +1,36 @@
+'''
+用于测试预测逻辑
+@Author: QST520
+'''
+import numpy as np
+
+import datetime
+import matplotlib.pyplot as plt
+
+
+def datetime_to_number(date: str):  # 格式化日期转换为 integer
+    date_number = datetime.datetime.strptime(date, "%Y-%m-%d")
+    base_number = datetime.datetime.strptime("2024-1-1", "%Y-%m-%d")
+    return date_number.__sub__(base_number).days
+
+
+if __name__ == '__main__':  # 预测 demo
+    xs = [
+        '2024-6-4', '2024-6-5', '2024-6-6', '2024-6-7', '2024-6-8', '2024-6-9',
+        '2024-6-10', '2024-6-11', '2024-6-12', '2024-6-13'
+    ]
+    ys = [15, 14, 16, 15, 16, 13, 12, 11, 9, 8]
+    xs = np.array(list(map(datetime_to_number, xs)))
+    ys = np.array(ys)
+    fit = np.polyfit(xs, ys, 1)
+    fn = np.poly1d(fit)
+    print('2024-6-14 PREDICTION: ' +
+          str(int(fn(datetime_to_number('2024-6-14')))))
+    print('2024-6-15 PREDICTION: ' +
+          str(int(fn(datetime_to_number('2024-6-15')))))
+    print('2024-6-16 PREDICTION: ' +
+          str(int(fn(datetime_to_number('2024-6-16')))))
+    print('2024-6-17 PREDICTION: ' +
+          str(int(fn(datetime_to_number('2024-6-17')))))
+    print('2024-6-18 PREDICTION: ' +
+          str(int(fn(datetime_to_number('2024-6-18')))))