如何将两个numpy数组转为DataFrame并合并得到指定格式结果
实现步骤
- 导入pandas库
import pandas as pd
- 用两个numpy数组直接构造目标格式DataFrame
# 替换为你自己的数组变量名即可 df = pd.DataFrame({ "Id": X, "SalePrice": y_test_pred })
- (可选)调整SalePrice列的小数位数,符合输出要求
# 示例为保留1位小数,可根据需求修改括号内的数值 df["SalePrice"] = df["SalePrice"].round(1)
- (可选)导出为CSV文件
# index=False参数会去掉默认生成的行索引,输出的CSV就和你期望的格式完全一致 df.to_csv("output.csv", index=False)
注意:操作前请确保X数组和y_test_pred数组的长度完全相同,否则会触发构造报错。
内容的提问来源于stack exchange,提问作者Dataleon
相关产品推荐
相关产品推荐

