理解np.apply_along_axis||Cython array-based functions

• Overview

  From ExponentialMovingWindow:

  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