pyvista.core._validation.check.check_ndim

Contents

pyvista.core._validation.check.check_ndim#

check_ndim(
array: _ArrayLikeOrScalar[NumberType],
/,
ndim: int | Sequence[int],
*,
name: str = 'Array',
)[source]#

Check if an array has the specified number of dimensions.

Note

Scalar values have a dimension of 0.

Parameters:
arrayfloat | ArrayLike[float]

Number or array to check.

ndimint | Sequence[int], optional

A single dimension or a sequence of allowable dimensions. If an integer, the array must have this number of dimension(s). If a sequence, the array must have at least one of the specified number of dimensions.

namestr, default: “Array”

Variable name to use in the error messages if any are raised.

Raises:
ValueError

If the array does not have the required number of dimensions.

Examples

Check if an array is one-dimensional

>>> import numpy as np
>>> from pyvista import _validation
>>> _validation.check_ndim([1, 2, 3], ndim=1)

Check if an array is two-dimensional or a scalar.

>>> _validation.check_ndim(1, ndim=(0, 2))