Function _apply_func2

Wrapping/Python/vtkmodules/numpy_interface/numpy_algorithms.py:14–27

Apply a function to each member of a VTKCompositeDataArray. Returns a list of arrays. Note that this function is mainly for internal use by this module.

(func, array, args)

from . import internal_algorithms as algs

def _apply_func2(func, array, args):
    """Apply a function to each member of a VTKCompositeDataArray.
    Returns a list of arrays.

    Note that this function is mainly for internal use by this module."""
    if array is dsa.NoneArray:
        return []
    res = []
    for a in array.Arrays:
        if a is dsa.NoneArray:
            res.append(dsa.NoneArray)
        else:
            res.append(func(a, *args))
    return res

def apply_ufunc(func, array, args=()):
    """Apply a function to each member of a VTKCompositeDataArray.