我正在进行一个机器学习项目，使用Python编写多元线性回归模型，以下是我的代码：

import matplotlib.pyplot as pltimport numpy as npimport pandas as pdfrom sklearn.linear_model import LinearRegressiondata = pd.read_csv("train.csv", delimiter=",", header=0)x = data['Col1'][:, np.newaxis]y = data['Expected']reg = LinearRegression()reg.fit(x, y)

我的train.csv文件包含三列：Col1、Col2和Expected。因此，x应该包含输入的”Col1″和”Col2″记录，而y应该包含输出”Expected”记录。我已经成功地将Col1作为输入放入x中，但无法将Col2也放入x中。

如何将Col1和Col2两列的值保存到x中，以便稍后用于线性回归拟合？

>>>> print(data.head())          Col1     Col2  Expected      0   7.645   5.2119      46.0      1   7.079   3.4145      28.7      2  91.900  24.0000      50.0      3   5.875   1.1296      50.0      4   6.153   3.2797      29.6

回答：

你需要做的就是：

x = data[['Col1', 'Col2']]