理解np.apply_along_axis||Cython array-based functions-CSDN博客

本文链接：https://blog.csdn.net/the_time_runner/article/details/108173283

Overview

def _apply(self, func):
    """
        Rolling statistical measure using supplied function. Designed to be
        used with passed-in Cython array-based functions.
        Parameters
        ----------
        func : str/callable to apply
        Returns
        -------
        y : same type as input argument
        """
    blocks, obj = self._create_blocks(self._selected_obj)
    block_list = list(blocks)

    results = []
    exclude = []
    for i, b in enumerate(blocks):
        try:
            values = self._prep_values(b.values)

            except (TypeError, NotImplementedError) as err:
                if isinstance(obj, ABCDataFrame):
                    exclude.extend(b.columns)
                    del block_list[i]
                    continue
                    else:
                        raise DataError("No numeric types to aggregate") from err

                        if values.size == 0:
                            results.append(values.copy())
                            continue

                            results.append(np.apply_along_axis(func, self.axis, values))

                            return self._wrap_results(results, block_list, obj, exclude)

np.apply_along_axis

numpy.apply_along_axis(func1d, axis, arr, *args, **kwargs)

Apply a function to 1-D slices along the given axis.

Execute func1d(a, *args, **kwargs) where func1d operator on 1-D arrays and a is a 1-D slice of arr along axis.

This is equivalent to (but faster than) the following use of ndindex and s_, which sets each of ii, jj and kk to a tuple of indices.
Cython array-based functions

Cython for NumPy users