ENH: changes requested in issue

Author: attack68

doc/source/whatsnew/v1.1.1.rst

@@ -51,11 +51,6 @@ Categorical
 
 -
 
-Other enhancements
-~~~~~~~~~~~~~~~~~~
-
-- :meth:`Styler.set_table_styles` now allows the direct styling of rows and columns and can be chained (:issue:`35607`)
-
 .. ---------------------------------------------------------------------------
 
 .. _whatsnew_111.contributors:

doc/source/whatsnew/v1.2.0.rst

@@ -18,6 +18,7 @@ Enhancements
 Other enhancements
 ^^^^^^^^^^^^^^^^^^
 
+- :meth:`Styler.set_table_styles` now allows the direct styling of rows and columns and can be chained (:issue:`35607`)
 -
 -
 

pandas/io/formats/style.py

@@ -933,15 +933,23 @@ def set_table_styles(self, table_styles, axis=0, overwrite=True) -> "Styler":
             in their respective tuple form. The dict values should be
             a list as specified in the form with CSS selectors and
             props that will be applied to the specified row or column.
+
+            .. versionchanged:: 1.2.0
+
         axis : {0 or 'index', 1 or 'columns', None}, default 0
             Apply to each column (``axis=0`` or ``'index'``), to each row
             (``axis=1`` or ``'columns'``). Only used if `table_styles` is
             dict.
+
+            .. versionadded:: 1.2.0
+
         overwrite : boolean, default True
             Styles are replaced if `True`, or extended if `False`. CSS
             rules are preserved so most recent styles set will dominate
             if selectors intersect.
 
+            .. versionadded:: 1.2.0
+
         Returns
         -------
         self : Styler
@@ -954,33 +962,38 @@ def set_table_styles(self, table_styles, axis=0, overwrite=True) -> "Styler":
         ...     [{'selector': 'tr:hover',
         ...       'props': [('background-color', 'yellow')]}]
         ... )
+
+        Adding column styling by name
+
         >>> df.style.set_table_styles({
         ...     'A': [{'selector': '',
         ...            'props': [('color', 'red')]}],
         ...     'B': [{'selector': 'td',
         ...            'props': [('color', 'blue')]}]
         ... }, overwrite=False)
+
+        Adding row styling
+
         >>> df.style.set_table_styles({
         ...     0: [{'selector': 'td:hover',
         ...          'props': [('font-size', '25px')]}]
         ... }, axis=1, overwrite=False)
         """
-        if isinstance(table_styles, dict):
-            if axis == 0 or axis == "index":
-                obj = self.data.columns
-                idf = ".col"
+        if is_dict_like(table_styles):
+            if axis in [0, "index"]:
+                obj, idf = self.data.columns, ".col"
             else:
-                obj = self.data.index
-                idf = ".row"
-
-            _styles = []
-            for key, styles in table_styles.items():
-                for s in styles:
-                    c = str(obj.get_loc(key))
-                    _styles.append(
-                        {"selector": s["selector"] + idf + c, "props": s["props"]}
-                    )
-            table_styles = _styles
+                obj, idf = self.data.index, ".row"
+
+            table_styles = [
+                {
+                    "selector": s["selector"] + idf + str(obj.get_loc(key)),
+                    "props": s["props"],
+                }
+                for key, styles in table_styles.items()
+                for s in styles
+            ]
+
         if not overwrite and self.table_styles is not None:
             self.table_styles.extend(table_styles)
         else:

pandas/tests/io/formats/test_style.py

@@ -1683,6 +1683,7 @@ def f(a, b, styler):
         assert result == (1, 2, styler)
 
     def test_chaining_table_styles(self):
+        # GH 35607
         df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
         styler = df.style.set_table_styles(
             [{"selector": "", "props": [("background-color", "yellow")]}]
@@ -1693,6 +1694,7 @@ def test_chaining_table_styles(self):
         assert len(styler.table_styles) == 2
 
     def test_column_and_row_styling(self):
+        # GH 35607
         df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
         s = Styler(df, uuid="_")
         s = s.set_table_styles({"A": [{"selector": "", "props": [("color", "blue")]}]})