sklearn.utils.validation.column_or_1d

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

Ravel column or 1d numpy array, else raises an error.

Parameters:
yarray-like

Input data.

dtypedata-type, default=None

Data type for y.

New in version 1.2.

warnbool, default=False

To control display of warnings.

Returns:
yndarray

Output data.

Raises:
ValueError

If y is not a 1D array or a 2D array with a single row or column.

Examples

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