column_or_1d#

sklearn.utils.validation.column_or_1d(y, *, dtype=None, warn=False, device=None)[源]#

展平列或一维 NumPy 数组,否则会引发错误。

参数:
y类数组

输入数据。

dtype数据类型, 默认为None

y 的数据类型。

在 1.2 版本中新增。

warn布尔值, 默认为False

控制警告的显示。

device设备, 默认为None

device 对象。更多详情请参见 Array API 用户指南

在 1.6 版本中新增。

返回:
yndarray

输出数据。

引发:
ValueError

如果 y 不是一维数组,也不是单行或单列的二维数组。

示例

>>> from sklearn.utils.validation import column_or_1d
>>> column_or_1d([1, 1])
array([1, 1])