# Examples from pyvista.DataSetFilters.remove_nan_cells
# =====================================================

# Create a small grid with some NaN point values and remove the
# affected cells.
import numpy as np
import pyvista as pv
grid = pv.ImageData(dimensions=(5, 5, 5))
values = np.arange(grid.n_points, dtype=float)
values[::7] = np.nan
grid.point_data['values'] = values
cleaned = grid.remove_nan_cells(scalars='values')
cleaned.n_cells < grid.n_cells
bool(np.any(np.isnan(cleaned.point_data['values'])))

# ----------------------------------------------------------------------
# Generated by sphinx-examples-as-code https://github.com/pyvista/sphinx-examples-as-code