python – 使用numpy 1d-array作为sklearn X的最短语法

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 使用numpy 1d-array作为sklearn X的最短语法，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含778字，纯文字阅读大概需要2分钟。

内容图文

我经常有两个numpy 1d数组,x和y,并且想要使用它们执行一些快速的sklearn拟合预测.

 import numpy as np
 from sklearn import linear_model

 # This is an example for the 1d aspect - it's obtained from something else.
 x = np.array([1, 3, 2, ...]) 
 y = np.array([12, 32, 4, ...])

现在我想做点什么

 linear_model.LinearRegression().fit(x, y)...

问题是它expects an X which is a 2d column array.因此,我通常喂它

 x.reshape((len(x), 1))

我觉得这很麻烦,难以阅读.

是否有一些更短的方法将1d数组转换为2d列数组(或者,或者sklearn接受1d数组)？

解决方法:

您可以切片阵列,创建一个newaxis：

x[:, None]

这个：

>>> x = np.arange(5)
>>> x[:, None]
array([[0],
       [1],
       [2],
       [3],
       [4]])

相当于：

>>> x.reshape(len(x), 1)
array([[0],
       [1],
       [2],
       [3],
       [4]])

如果您发现它更具可读性,则可以使用转置矩阵：

np.matrix(x).T

如果你想要一个数组：

np.matrix(x).T.A

内容总结

以上是互联网集市为您收集整理的python – 使用numpy 1d-array作为sklearn X的最短语法全部内容，希望文章能够帮你解决python – 使用numpy 1d-array作为sklearn X的最短语法所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端