faster Axis.apply

gdementen · gdementen · commit 3d68c53a7462 · 2019-06-26T17:04:58.000+02:00
diff --git a/doc/source/changes/version_0_30.rst.inc b/doc/source/changes/version_0_30.rst.inc
@@ -138,10 +138,6 @@ New features
 
 * implemented :py:obj:`Axis.apply()` method to transform an axis labels by a function and return a new Axis.
 
-    >>> sex = Axis('sex=MALE,FEMALE')
-    >>> sex.apply(str.capitalize)
-    Axis(['Male', 'Female'], 'sex')
-
 * added :py:obj:`Session.update()` method to add and modify items from an existing session by passing
   either another session or a dict-like object or an iterable object with (key, value) pairs (closes :issue:`754`).
 
diff --git a/larray/core/axis.py b/larray/core/axis.py
@@ -19,6 +19,9 @@
                               renamed_to, common_type, LHDFStore, lazy_attribute, _isnoneslice, unique_multi, Product)
 
 
+np_frompyfunc = np.frompyfunc
+
+
 class Axis(ABCAxis):
     """
     Represents an axis. It consists of a name and a list of labels.
@@ -1105,7 +1108,7 @@ def apply(self, func):
         >>> sex.apply(str.capitalize)
         Axis(['Male', 'Female'], 'sex')
         """
-        return Axis(map(func, self.labels), self.name)
+        return Axis(np_frompyfunc(func, 1, 1)(self.labels), self.name)
 
     # XXX: rename to named like Group?
     def rename(self, name):
