股票预测绘图索引问题：如何将预测结果转为%Y%m%d日期索引DataFrame

解决方案 #

直接复用测试集test_df的日期索引，转换为目标格式后赋值给预测结果即可，操作如下：

前提 #

按时间顺序做8:2时序分割得到的test_df共252条记录，和预测结果的顺序一一对应、长度完全匹配。

代码实现 #

情况1：预测结果为numpy数组/列表/带整数索引的Series

import pandas as pd

# 替换pred为你实际的预测结果变量
pred_df = pd.DataFrame(
    pred,
    columns=['预测值'], # 可自定义列名
    index=test_df.index.strftime('%Y%m%d') # 直接复用测试集日期，转换为%Y%m%d字符串格式作为索引
)

情况2：预测结果已经是带整数索引的DataFrame

直接修改索引即可：

pred_df = pred.set_index(test_df.index.strftime('%Y%m%d'))

补充说明 #

如果你需要索引本身保持datetime类型，仅展示时按照%Y%m%d格式显示，可以设置pandas全局日期显示格式：

pd.set_option('display.date_format', '%Y%m%d')
pred_df = pd.DataFrame(pred, columns=['预测值'], index=test_df.index)

内容的提问来源于stack exchange，提问作者Wind