您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

将Pandas数据框转换为数组并评估多元线性回归模型

将Pandas数据框转换为数组并评估多元线性回归模型

您可以as_matrix直接在数据框对象上使用方法将数据框转换为矩阵。您可能需要指定感兴趣的列,X=df[['x1','x2','X3']].as_matrix()其中不同的x是列名。

对于y变量,您可以y = df['ground_truth'].values用来获取数组。

这是一些随机生成的数据的示例:

import numpy as np
#create a 5X5 dataframe
df = pd.DataFrame(np.random.random_integers(0, 100, (5, 5)), columns = ['X1','X2','X3','X4','y'])

调用as_matrix()df返回一个numpy.ndarray对象

X = df[['X1','X2','X3','X4']].as_matrix()

调用从熊猫values返回anumpy.ndarray``series

y =df['y'].values

注意:您可能会收到一条警告:FutureWarning: Method .as_matrix will be removed in a future version. Use .values instead.

要修复它,请使用values而不是as_matrix如下所示

X = df[['X1','X2','X3','X4']].values
其他 2022/1/1 18:29:04 有509人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶