ENH: applymap get kwargs pandas-dev#39987

Author: alexprincel

pandas/core/frame.py

@@ -13,6 +13,7 @@
 import collections
 from collections import abc
 import datetime
+import functools
 from io import StringIO
 import itertools
 import mmap
@@ -3045,7 +3046,6 @@ def _ixs(self, i: int, axis: int = 0):
 
             # this is a cached value, mark it so
             result._set_as_cached(label, self)
-
             return result
 
     def _get_column_array(self, i: int) -> ArrayLike:
@@ -7934,7 +7934,7 @@ def apply(
         return op.apply()
 
     def applymap(
-        self, func: PythonFuncType, na_action: Optional[str] = None
+        self, func: PythonFuncType, na_action: Optional[str] = None, **kwargs
     ) -> DataFrame:
         """
         Apply a function to a Dataframe elementwise.
@@ -7948,6 +7948,9 @@ def applymap(
             Python function, returns a single value from a single value.
         na_action : {None, 'ignore'}, default None
             If ‘ignore’, propagate NaN values, without passing them to func.
+        **kwargs
+            Additional keyword arguments to pass as keywords arguments to
+            `func`.
 
             .. versionadded:: 1.2
 
@@ -8002,6 +8005,7 @@ def applymap(
                 f"na_action must be 'ignore' or None. Got {repr(na_action)}"
             )
         ignore_na = na_action == "ignore"
+        func = functools.partial(func, **kwargs)
 
         # if we have a dtype == 'M8[ns]', provide boxed values
         def infer(x):

pandas/tests/apply/test_frame_apply.py

@@ -678,6 +678,12 @@ def test_applymap(float_frame):
             tm.assert_frame_equal(result, frame)
 
 
+def test_applymap_kwargs():
+    result = pd.DataFrame([[1, 2], [3, 4]]).applymap(lambda x, y: x + y, y=2)
+    expected = pd.DataFrame([[3, 4], [5, 6]])
+    assert tm.assert_frame_equal(result, expected)
+
+
 def test_applymap_na_ignore(float_frame):
     # GH 23803
     strlen_frame = float_frame.applymap(lambda x: len(str(x)))