如何在Python中将Random Forest模型的predict结果转为列表或DataFrame

没问题，这两种转换操作都非常直接，我给你分别演示具体实现方式：

转换为Python列表 #

NumPy数组自带了tolist()方法，可以一键转换成标准的Python列表，完全不需要额外处理：

# 假设你的预测结果是numpy数组prediction
prediction_list = prediction.tolist()
# 输出就是普通列表：[1, 1, 1, 1, 0, 0, ...]

转换为Pandas DataFrame #

如果需要转换成表格形式方便后续分析或导出，可以用Pandas的DataFrame构造函数，还能自定义列名让结果更直观：

import pandas as pd

# 转换为DataFrame，指定列名为"预测标签"
prediction_df = pd.DataFrame(prediction, columns=["预测标签"])

# 查看前几行结果
print(prediction_df.head())

运行后会得到一个带列名的表格，每一行对应一个预测值，非常适合做后续的统计、可视化或者导出成CSV/Excel文件。

如果你的预测结果是多分类的二维数组（比如每个样本对应多个类别概率），只需要调整列名即可，比如：

# 假设是3分类的概率输出，列名对应三个类别
prediction_df = pd.DataFrame(prediction, columns=["类别A概率", "类别B概率", "类别C概率"])

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