Function apply_ufunc

Wrapping/Python/vtkmodules/numpy_interface/numpy_algorithms.py:29–37

Apply a function to each member of a VTKCompositeDataArray. VTKArray and numpy arrays are also supported.

(func, array, args=())

    return res

def apply_ufunc(func, array, args=()):
    """Apply a function to each member of a VTKCompositeDataArray.
    VTKArray and numpy arrays are also supported."""
    if array is dsa.NoneArray:
        return dsa.NoneArray
    elif type(array) == dsa.VTKCompositeDataArray:
        return dsa.VTKCompositeDataArray(_apply_func2(func, array, args), dataset = array.DataSet)
    else:
        return func(array, *args)

def _make_ufunc(ufunc):
    """ Given a ufunc, creates a closure that applies it to each member