能给我提一个python代码吗? 我现在有这样一个训练数据它提供了2015 年 9 月 1 日至 2018 年 12 月 20 日的数据,其存储在(order_train1.xlsx)文件中,包含以下信息(冒号后面为样例,是数据格式)。 order_date(订单日期):2015/9/1; sales_region_code(销 售区域编码):104; item_code(产品编码):22069; ord_qty (订单需求量):19; 现在要用使用随机森林法则对未来三个月的数据进行预测。 然后,将结果保存在(predict_sku1.xlsx)文件中,其格式为(冒号后面为样例,是数据格式)。 sales_region_code:101; item_code:20002; 2019 年 1 月预 测需求量:1; 2019 年 2 月预 测需求量:3; 2019 年 3 月预 测需求量:2;
时间: 2023-06-05 14:06:06 浏览: 99
好的,以下是一个简单的Python代码:
```python
import pandas as pd
from sklearn.ensemble import RandomForestRegressor
# 加载数据
train_df = pd.read_excel('order_train1.xlsx')
# 处理数据
train_df.drop(['order_date'], axis=1, inplace=True) # 去掉订单日期列
X = train_df.drop(['ord_qty'], axis=1) # 特征
y = train_df['ord_qty'] # 目标值
# 训练模型
model = RandomForestRegressor(n_estimators=100, random_state=42)
model.fit(X, y)
# 预测未来三个月的数据
prediction_df = pd.read_excel('predict_sku1.xlsx')
X_test = prediction_df.drop(['2019 年 1 月预测需求量', '2019 年 2 月预测需求量', '2019 年 3 月预测需求量'], axis=1)
y_predict = model.predict(X_test)
# 保存结果
prediction_df['2019 年 1 月预测需求量'] = y_predict[0]
prediction_df['2019 年 2 月预测需求量'] = y_predict[1]
prediction_df['2019 年 3 月预测需求量'] = y_predict[2]
prediction_df.to_excel('predict_sku1.xlsx', index=False)
```
请确认我理解您的问题。您希望使用随机森林算法预测未来三个月的销量,并将结果保存在一个 Excel 文件中。您已经有了一个包含训练数据的 Excel 文件(`order_train1.xlsx`),其中包含了 2015 年 9 月 1 日至 2018 年 12 月 20 日的销售数据。您还有一个 Excel 文件(`predict_sku1.xlsx`),其中包含了需要预测的销售地区和商品编码。
如果我有任何错误或对问题的理解不够准确,请告诉我。
阅读全文