如何将多个多维数组保存至同一个CSV文件并自定义列名

修改方案 #

你可以直接合并两个二维数组后指定列名生成DataFrame，完整修改后的代码如下：

import pandas as pd
from sklearn.datasets import make_regression

# 调整n_targets参数为2，生成形状为(10,2)的y
X, y = make_regression(n_samples=10, n_features=2, n_informative=2, n_targets=2, random_state=1, noise=0.5)

# 构造DataFrame时指定X的列名
my_df = pd.DataFrame(X, columns=["X1", "X2"])
# 新增y的两列并指定列名
my_df[["y1", "y2"]] = y

# 导出为CSV文件
my_df.to_csv('test_data.csv', index=False, header=True)

你也可以使用numpy的水平拼接方法一次性构造DataFrame，效果完全一致：

import numpy as np
import pandas as pd
from sklearn.datasets import make_regression

X, y = make_regression(n_samples=10, n_features=2, n_informative=2, n_targets=2, random_state=1, noise=0.5)
# 水平拼接X和y后直接指定全部列名
my_df = pd.DataFrame(np.hstack([X, y]), columns=["X1", "X2", "y1", "y2"])
my_df.to_csv('test_data.csv', index=False, header=True)

运行后导出的CSV就会包含X1、X2、y1、y2四列，和预期输出格式匹配。

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