From af99333fe5fe9cdd0f446d12e46306ff9490290c Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Fri, 7 Aug 2020 11:58:43 +0200
Subject: [PATCH 01/11] ENH: add named based column css styling to pandas
 Styler

---
 doc/source/user_guide/style.ipynb     | 44 +++++++++++++--
 pandas/io/formats/style.py            | 80 +++++++++++++++++++++++++++
 pandas/tests/io/formats/test_style.py | 22 ++++++++
 3 files changed, 141 insertions(+), 5 deletions(-)

diff --git a/doc/source/user_guide/style.ipynb b/doc/source/user_guide/style.ipynb
index 77a1fef28f373..460eef0b19508 100644
--- a/doc/source/user_guide/style.ipynb
+++ b/doc/source/user_guide/style.ipynb
@@ -793,7 +793,8 @@
    "source": [
     "The next option you have are \"table styles\".\n",
     "These are styles that apply to the table as a whole, but don't look at the data.\n",
-    "Certain sytlings, including pseudo-selectors like `:hover` can only be used this way."
+    "Certain stylings, including pseudo-selectors like `:hover` can only be used this way.\n",
+    "These can also be used to set specific row or column based class selectors, as will be shown."
    ]
   },
   {
@@ -831,9 +832,31 @@
     "The value for `props` should be a list of tuples of `('attribute', 'value')`.\n",
     "\n",
     "`table_styles` are extremely flexible, but not as fun to type out by hand.\n",
-    "We hope to collect some useful ones either in pandas, or preferable in a new package that [builds on top](#Extensibility) the tools here."
+    "We hope to collect some useful ones either in pandas, or preferable in a new package that [builds on top](#Extensibility) the tools here.\n",
+    "\n",
+    "`table_styles` can be used to add column and row based class descriptors. For large tables this can increase performance by avoiding repetitive individual css for each cell, and it can also simplify style construction in some cases.\n",
+    "`Styler.extend_column_styles` is a specific function to set styles to individual columns.\n",
+    "\n",
+    "Note that `Styler.set_table_styles` will overwrite existing styles (including those column based) whilst the preferred `Styler.extend_table_styles` will add to the current items."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [
+    "html = html.extend_column_styles({\n",
+    "    'B': [dict(selector='', props=[('color', 'blue')])]\n",
+    "})\n",
+    "html"
+   ],
+   "metadata": {
+    "collapsed": false,
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   }
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -922,10 +945,12 @@
     "- DataFrame only `(use Series.to_frame().style)`\n",
     "- The index and columns must be unique\n",
     "- No large repr, and performance isn't great; this is intended for summary DataFrames\n",
-    "- You can only style the *values*, not the index or columns\n",
+    "- You can only style the *values*, not the index or columns (except with `table_styles` above)\n",
     "- You can only apply styles, you can't insert new HTML entities\n",
     "\n",
-    "Some of these will be addressed in the future.\n"
+    "Some of these will be addressed in the future.\n",
+    "Performance can suffer when adding styles to each cell in a large DataFrame.\n",
+    "It is recommended to apply table or column based styles where possible to limit overall HTML length, as well as setting a shorter UUID to avoid unnecessary repeated data transmission. \n"
    ]
   },
   {
@@ -1237,8 +1262,17 @@
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
    "version": "3.7.0"
+  },
+  "pycharm": {
+   "stem_cell": {
+    "cell_type": "raw",
+    "source": [],
+    "metadata": {
+     "collapsed": false
+    }
+   }
   }
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}
\ No newline at end of file
diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index 3bbb5271bce61..86169cdc8c8dd 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -939,6 +939,86 @@ def set_table_styles(self, table_styles) -> "Styler":
         self.table_styles = table_styles
         return self
 
+    def extend_table_styles(self, table_styles) -> "Styler":
+        """
+        Extend the existing table styles on a Styler.
+
+        These are placed in a ``<style>`` tag before the generated HTML table.
+
+        Parameters
+        ----------
+        table_styles : list
+            Each individual table_style should be a dictionary with
+            ``selector`` and ``props`` keys. ``selector`` should be a CSS
+            selector that the style will be applied to (automatically
+            prefixed by the table's UUID) and ``props`` should be a list of
+            tuples with ``(attribute, value)``.
+
+        Returns
+        -------
+        self : Styler
+
+        Examples
+        --------
+        >>> df = pd.DataFrame(np.random.randn(10, 4))
+        >>> df.style.set_table_styles(
+        ...     [{'selector': 'tr:hover',
+        ...       'props': [('background-color', 'yellow')]}]
+        ... ).extend_table_styles(
+        ...     [{'selector': '.col2',
+        ...       'props': [('background-color', 'blue')]}]
+        ...)
+        """
+        if self.table_styles is None:
+            return self.set_table_styles(table_styles)
+        self.table_styles.extend(table_styles)
+        return self
+
+    def extend_column_styles(self, column_styles) -> "Styler":
+        """
+        Sets class styles for each column on a Styler.
+
+        These are placed in a ``<style>`` tag before the generated HTML table.
+
+        Parameters
+        ----------
+        column_styles : dict
+            Each key of the dict should be a column header with the value
+            being a list of individual styles. Each style is a dictionary with
+            ``selector`` and ``props`` keys. ``selector`` should be a CSS
+            selector that the style will be applied to (automatically
+            prefixed by the table's UUID) and ``props`` should be a list of
+            tuples with ``(attribute, value)``.
+
+        Returns
+        -------
+        self : Styler
+
+        Notes
+        -----
+        Where the ``selector`` is an empty string or `None` the ``props`` will
+        be applied to the column class generically.
+
+        Examples
+        --------
+        >>> df = pd.DataFrame(np.random.randn(3, 2), columns=['a', 'b'])
+        >>> df.style.extend_column_styles({
+        ...        'a': [{'selector': '',
+        ...               'props': [('font-size', '20px')]},
+        ...              {'selector': 'td:hover',
+        ...               'props': [('color', 'pink')]}],
+        ...        'b': [{'selector': '',
+        ...               'props': [('font-size', '10px')]}]
+        ... })
+        """
+        _styles = []
+        for col, styles in column_styles.items():
+            for s in styles:
+                c = str(self.data.columns.get_loc(col))
+                _styles.append({'selector': s['selector'] + '.col' + c,
+                                'props': s['props']})
+        return self.extend_table_styles(_styles)
+
     def set_na_rep(self, na_rep: str) -> "Styler":
         """
         Set the missing data representation on a Styler.
diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index 6025649e9dbec..4cd39ac207b41 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1691,6 +1691,28 @@ def test_no_cell_ids(self):
         s = styler.render()  # render twice to ensure ctx is not updated
         assert s.find('<td  class="data row0 col0" >') != -1
 
+    def test_extending_table_styles(self):
+        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=['A', 'B'])
+        styler = df.style.set_table_styles([{
+            'selector': '',
+            'props': [('background-color', 'yellow')]}]
+        ).extend_table_styles([{
+            'selector': '.col0',
+            'props': [('background-color', 'blue')]}]
+        )
+        assert len(styler.table_styles) == 2
+
+    def test_column_styling(self):
+        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=['A', 'B'])
+        s = Styler(df, uuid='_')
+        s = s.extend_column_styles({
+            'A': [{
+                'selector': '',
+                'props': [('color', 'blue')]
+            }]
+        })
+        assert '#T__ .col0 {\n          color: blue;\n    }' in s.render()
+
 
 @td.skip_if_no_mpl
 class TestStylerMatplotlibDep:

From c8097edc66515fc2f0247c01de022eff88d5b400 Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Fri, 7 Aug 2020 12:34:02 +0200
Subject: [PATCH 02/11] ENH: add named based column css styling to pandas
 Styler (black fixup)

---
 pandas/tests/io/formats/test_style.py | 27 +++++++++++----------------
 1 file changed, 11 insertions(+), 16 deletions(-)

diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index 4cd39ac207b41..becae9b9922b5 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1692,26 +1692,21 @@ def test_no_cell_ids(self):
         assert s.find('<td  class="data row0 col0" >') != -1
 
     def test_extending_table_styles(self):
-        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=['A', 'B'])
-        styler = df.style.set_table_styles([{
-            'selector': '',
-            'props': [('background-color', 'yellow')]}]
-        ).extend_table_styles([{
-            'selector': '.col0',
-            'props': [('background-color', 'blue')]}]
+        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
+        styler = df.style.set_table_styles(
+            [{"selector": "", "props": [("background-color", "yellow")]}]
+        ).extend_table_styles(
+            [{"selector": ".col0", "props": [("background-color", "blue")]}]
         )
         assert len(styler.table_styles) == 2
 
     def test_column_styling(self):
-        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=['A', 'B'])
-        s = Styler(df, uuid='_')
-        s = s.extend_column_styles({
-            'A': [{
-                'selector': '',
-                'props': [('color', 'blue')]
-            }]
-        })
-        assert '#T__ .col0 {\n          color: blue;\n    }' in s.render()
+        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
+        s = Styler(df, uuid="_")
+        s = s.extend_column_styles(
+            {"A": [{"selector": "", "props": [("color", "blue")]}]}
+        )
+        assert "#T__ .col0 {\n          color: blue;\n    }" in s.render()
 
 
 @td.skip_if_no_mpl

From 07591da3f05e4bdec0f9d54bc3fa7dfb09fb6d0e Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Fri, 7 Aug 2020 13:21:32 +0200
Subject: [PATCH 03/11] ENH: add named based column css styling to pandas
 Styler (black fixup)

---
 pandas/io/formats/style.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index 86169cdc8c8dd..ed09f0a9cb961 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -1015,8 +1015,9 @@ def extend_column_styles(self, column_styles) -> "Styler":
         for col, styles in column_styles.items():
             for s in styles:
                 c = str(self.data.columns.get_loc(col))
-                _styles.append({'selector': s['selector'] + '.col' + c,
-                                'props': s['props']})
+                _styles.append(
+                    {"selector": s["selector"] + ".col" + c, "props": s["props"]}
+                )
         return self.extend_table_styles(_styles)
 
     def set_na_rep(self, na_rep: str) -> "Styler":

From d14382446ab3d1701860a195b591f699d0ca6f5d Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Fri, 7 Aug 2020 16:39:38 +0200
Subject: [PATCH 04/11] ENH: add named based column css styling to pandas
 Styler (black fixup)

---
 doc/source/whatsnew/v1.1.1.rst | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/doc/source/whatsnew/v1.1.1.rst b/doc/source/whatsnew/v1.1.1.rst
index 77ea67f76f655..b318b7042faea 100644
--- a/doc/source/whatsnew/v1.1.1.rst
+++ b/doc/source/whatsnew/v1.1.1.rst
@@ -46,6 +46,12 @@ Bug fixes
 - Bug in ``.groupby(..).rolling(..)`` where passing ``closed`` with column selection would raise a ``ValueError`` (:issue:`35549`)
 - Bug in :class:`DataFrame` constructor failing to raise ``ValueError`` in some cases when ``data`` and ``index`` have mismatched lengths (:issue:`33437`)
 
+Other enhancements
+~~~~~~~~~~~~~~~~~~
+
+- :meth:`Styler.extend_table_styles` added the supplement `Styler.set_table_styles` (:issue:`35607`)
+- :meth:`Styler.extend_column_styles` to directly add class based CSS to specified DataFrame columns (:issue:`35607`)
+
 .. ---------------------------------------------------------------------------
 
 .. _whatsnew_111.contributors:

From a4ebd1a7abe1b12f6c5e4cd53f032ac4a2ffbe9f Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Fri, 7 Aug 2020 20:51:52 +0200
Subject: [PATCH 05/11] ENH: restructure code for no new functions, only
 keyword additions

---
 doc/source/user_guide/style.ipynb     |  11 +-
 doc/source/whatsnew/v1.1.1.rst        |   3 +-
 pandas/io/formats/style.py            | 149 ++++++++++----------------
 pandas/tests/io/formats/test_style.py |  16 ++-
 4 files changed, 76 insertions(+), 103 deletions(-)

diff --git a/doc/source/user_guide/style.ipynb b/doc/source/user_guide/style.ipynb
index 460eef0b19508..80ed8be90928e 100644
--- a/doc/source/user_guide/style.ipynb
+++ b/doc/source/user_guide/style.ipynb
@@ -835,9 +835,9 @@
     "We hope to collect some useful ones either in pandas, or preferable in a new package that [builds on top](#Extensibility) the tools here.\n",
     "\n",
     "`table_styles` can be used to add column and row based class descriptors. For large tables this can increase performance by avoiding repetitive individual css for each cell, and it can also simplify style construction in some cases.\n",
-    "`Styler.extend_column_styles` is a specific function to set styles to individual columns.\n",
+    "If `table_styles` is given as a dictionary each key should be a specified column or index value and this will map to specific class CSS selectors of the given column or row.\n",
     "\n",
-    "Note that `Styler.set_table_styles` will overwrite existing styles (including those column based) whilst the preferred `Styler.extend_table_styles` will add to the current items."
+    "Note that `Styler.set_table_styles` will overwrite existing styles but can be chained by setting the `overwrite` argument to `False`."
    ]
   },
   {
@@ -845,9 +845,10 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "html = html.extend_column_styles({\n",
-    "    'B': [dict(selector='', props=[('color', 'blue')])]\n",
-    "})\n",
+    "html = html.set_table_styles({\n",
+    "    'B': [dict(selector='', props=[('color', 'green')])],\n",
+    "    'C': [dict(selector='td', props=[('color', 'red')])], \n",
+    "    }, overwrite=False)\n",
     "html"
    ],
    "metadata": {
diff --git a/doc/source/whatsnew/v1.1.1.rst b/doc/source/whatsnew/v1.1.1.rst
index b318b7042faea..c6fae2dbe8b55 100644
--- a/doc/source/whatsnew/v1.1.1.rst
+++ b/doc/source/whatsnew/v1.1.1.rst
@@ -49,8 +49,7 @@ Bug fixes
 Other enhancements
 ~~~~~~~~~~~~~~~~~~
 
-- :meth:`Styler.extend_table_styles` added the supplement `Styler.set_table_styles` (:issue:`35607`)
-- :meth:`Styler.extend_column_styles` to directly add class based CSS to specified DataFrame columns (:issue:`35607`)
+- :meth:`Styler.set_table_styles` now allows the direct styling of rows and columns and can be chained (:issue:`35607`)
 
 .. ---------------------------------------------------------------------------
 
diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index ed09f0a9cb961..99b2137b012b0 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -909,50 +909,38 @@ def set_caption(self, caption: str) -> "Styler":
         self.caption = caption
         return self
 
-    def set_table_styles(self, table_styles) -> "Styler":
+    def set_table_styles(self, table_styles, axis=0, overwrite=True) -> "Styler":
         """
         Set the table styles on a Styler.
 
         These are placed in a ``<style>`` tag before the generated HTML table.
 
-        Parameters
-        ----------
-        table_styles : list
-            Each individual table_style should be a dictionary with
-            ``selector`` and ``props`` keys. ``selector`` should be a CSS
-            selector that the style will be applied to (automatically
-            prefixed by the table's UUID) and ``props`` should be a list of
-            tuples with ``(attribute, value)``.
-
-        Returns
-        -------
-        self : Styler
-
-        Examples
-        --------
-        >>> df = pd.DataFrame(np.random.randn(10, 4))
-        >>> df.style.set_table_styles(
-        ...     [{'selector': 'tr:hover',
-        ...       'props': [('background-color', 'yellow')]}]
-        ... )
-        """
-        self.table_styles = table_styles
-        return self
-
-    def extend_table_styles(self, table_styles) -> "Styler":
-        """
-        Extend the existing table styles on a Styler.
-
-        These are placed in a ``<style>`` tag before the generated HTML table.
+        This function can be used to style the entire table, columns, rows or
+        specific HTML selectors.
 
         Parameters
         ----------
-        table_styles : list
-            Each individual table_style should be a dictionary with
-            ``selector`` and ``props`` keys. ``selector`` should be a CSS
-            selector that the style will be applied to (automatically
-            prefixed by the table's UUID) and ``props`` should be a list of
-            tuples with ``(attribute, value)``.
+        table_styles : list or dict
+            If supplying a list, each individual table_style should be a
+            dictionary with ``selector`` and ``props`` keys. ``selector``
+            should be a CSS selector that the style will be applied to
+            (automatically prefixed by the table's UUID) and ``props``
+            should be a list of tuples with ``(attribute, value)``.
+            If supplying a dict, the dict keys should correspond to
+            column names or index values, depending upon the specified
+            `axis` argument. These will be mapped to row or col CSS
+            selectors. MultiIndex values as dict keys should be
+            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.
+        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.
+        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.
 
         Returns
         -------
@@ -960,66 +948,45 @@ def extend_table_styles(self, table_styles) -> "Styler":
 
         Examples
         --------
-        >>> df = pd.DataFrame(np.random.randn(10, 4))
+        >>> df = pd.DataFrame(np.random.randn(10, 4),
+        ...                   columns=['A', 'B', 'C', 'D'])
         >>> df.style.set_table_styles(
         ...     [{'selector': 'tr:hover',
         ...       'props': [('background-color', 'yellow')]}]
-        ... ).extend_table_styles(
-        ...     [{'selector': '.col2',
-        ...       'props': [('background-color', 'blue')]}]
-        ...)
-        """
-        if self.table_styles is None:
-            return self.set_table_styles(table_styles)
-        self.table_styles.extend(table_styles)
+        ... )
+        >>> df.style.set_table_styles({
+        ...     'A': [{'selector': '',
+        ...            'props': [('color', 'red')]}],
+        ...     'B': [{'selector': 'td',
+        ...            'props': [('color', 'blue')]}]
+        ... }, overwrite=False)
+        >>> 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'
+            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
+        if not overwrite and self.table_styles is not None:
+            self.table_styles.extend(table_styles)
+        else:
+            self.table_styles = table_styles
         return self
 
-    def extend_column_styles(self, column_styles) -> "Styler":
-        """
-        Sets class styles for each column on a Styler.
-
-        These are placed in a ``<style>`` tag before the generated HTML table.
-
-        Parameters
-        ----------
-        column_styles : dict
-            Each key of the dict should be a column header with the value
-            being a list of individual styles. Each style is a dictionary with
-            ``selector`` and ``props`` keys. ``selector`` should be a CSS
-            selector that the style will be applied to (automatically
-            prefixed by the table's UUID) and ``props`` should be a list of
-            tuples with ``(attribute, value)``.
-
-        Returns
-        -------
-        self : Styler
-
-        Notes
-        -----
-        Where the ``selector`` is an empty string or `None` the ``props`` will
-        be applied to the column class generically.
-
-        Examples
-        --------
-        >>> df = pd.DataFrame(np.random.randn(3, 2), columns=['a', 'b'])
-        >>> df.style.extend_column_styles({
-        ...        'a': [{'selector': '',
-        ...               'props': [('font-size', '20px')]},
-        ...              {'selector': 'td:hover',
-        ...               'props': [('color', 'pink')]}],
-        ...        'b': [{'selector': '',
-        ...               'props': [('font-size', '10px')]}]
-        ... })
-        """
-        _styles = []
-        for col, styles in column_styles.items():
-            for s in styles:
-                c = str(self.data.columns.get_loc(col))
-                _styles.append(
-                    {"selector": s["selector"] + ".col" + c, "props": s["props"]}
-                )
-        return self.extend_table_styles(_styles)
-
     def set_na_rep(self, na_rep: str) -> "Styler":
         """
         Set the missing data representation on a Styler.
diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index becae9b9922b5..fcfa94f94a406 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1691,22 +1691,28 @@ def test_no_cell_ids(self):
         s = styler.render()  # render twice to ensure ctx is not updated
         assert s.find('<td  class="data row0 col0" >') != -1
 
-    def test_extending_table_styles(self):
+    def test_chaining_table_styles(self):
         df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
         styler = df.style.set_table_styles(
             [{"selector": "", "props": [("background-color", "yellow")]}]
-        ).extend_table_styles(
-            [{"selector": ".col0", "props": [("background-color", "blue")]}]
+        ).set_table_styles(
+            [{"selector": ".col0", "props": [("background-color", "blue")]}],
+            overwrite=False
         )
         assert len(styler.table_styles) == 2
 
-    def test_column_styling(self):
+    def test_column_and_row_styling(self):
         df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
         s = Styler(df, uuid="_")
-        s = s.extend_column_styles(
+        s = s.set_table_styles(
             {"A": [{"selector": "", "props": [("color", "blue")]}]}
         )
         assert "#T__ .col0 {\n          color: blue;\n    }" in s.render()
+        s = s.set_table_styles(
+            {0: [{"selector": "", "props": [("color", "blue")]}]},
+            axis=1
+        )
+        assert "#T__ .row0 {\n          color: blue;\n    }" in s.render()
 
 
 @td.skip_if_no_mpl

From cae9159b8811336b1e643b1314d59a4ecb309657 Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Fri, 7 Aug 2020 20:52:51 +0200
Subject: [PATCH 06/11] ENH: black fix

---
 pandas/io/formats/style.py            | 6 +++---
 pandas/tests/io/formats/test_style.py | 9 +++------
 2 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index 99b2137b012b0..c8e1b0f858ef6 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -966,12 +966,12 @@ def set_table_styles(self, table_styles, axis=0, overwrite=True) -> "Styler":
         ... }, axis=1, overwrite=False)
         """
         if isinstance(table_styles, dict):
-            if axis == 0 or axis == 'index':
+            if axis == 0 or axis == "index":
                 obj = self.data.columns
-                idf = '.col'
+                idf = ".col"
             else:
                 obj = self.data.index
-                idf = '.row'
+                idf = ".row"
 
             _styles = []
             for key, styles in table_styles.items():
diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index fcfa94f94a406..79cf4526fa0c4 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1697,20 +1697,17 @@ def test_chaining_table_styles(self):
             [{"selector": "", "props": [("background-color", "yellow")]}]
         ).set_table_styles(
             [{"selector": ".col0", "props": [("background-color", "blue")]}],
-            overwrite=False
+            overwrite=False,
         )
         assert len(styler.table_styles) == 2
 
     def test_column_and_row_styling(self):
         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")]}]}
-        )
+        s = s.set_table_styles({"A": [{"selector": "", "props": [("color", "blue")]}]})
         assert "#T__ .col0 {\n          color: blue;\n    }" in s.render()
         s = s.set_table_styles(
-            {0: [{"selector": "", "props": [("color", "blue")]}]},
-            axis=1
+            {0: [{"selector": "", "props": [("color", "blue")]}]}, axis=1
         )
         assert "#T__ .row0 {\n          color: blue;\n    }" in s.render()
 

From 08fc8648d094902be2063cb91332db5e34112fb2 Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Sat, 8 Aug 2020 08:23:43 +0200
Subject: [PATCH 07/11] ENH: changes requested in issue

---
 doc/source/whatsnew/v1.1.1.rst        |  5 ----
 doc/source/whatsnew/v1.2.0.rst        |  1 +
 pandas/io/formats/style.py            | 43 +++++++++++++++++----------
 pandas/tests/io/formats/test_style.py |  2 ++
 4 files changed, 31 insertions(+), 20 deletions(-)

diff --git a/doc/source/whatsnew/v1.1.1.rst b/doc/source/whatsnew/v1.1.1.rst
index c6fae2dbe8b55..77ea67f76f655 100644
--- a/doc/source/whatsnew/v1.1.1.rst
+++ b/doc/source/whatsnew/v1.1.1.rst
@@ -46,11 +46,6 @@ Bug fixes
 - Bug in ``.groupby(..).rolling(..)`` where passing ``closed`` with column selection would raise a ``ValueError`` (:issue:`35549`)
 - Bug in :class:`DataFrame` constructor failing to raise ``ValueError`` in some cases when ``data`` and ``index`` have mismatched lengths (:issue:`33437`)
 
-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:
diff --git a/doc/source/whatsnew/v1.2.0.rst b/doc/source/whatsnew/v1.2.0.rst
index 55570341cf4e8..97cd71ce6d842 100644
--- a/doc/source/whatsnew/v1.2.0.rst
+++ b/doc/source/whatsnew/v1.2.0.rst
@@ -54,6 +54,7 @@ For example:
 Other enhancements
 ^^^^^^^^^^^^^^^^^^
 - :class:`Index` with object dtype supports division and multiplication (:issue:`34160`)
+- :meth:`Styler.set_table_styles` now allows the direct styling of rows and columns and can be chained (:issue:`35607`)
 -
 -
 
diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index c8e1b0f858ef6..5e95a25da56f9 100644
--- a/pandas/io/formats/style.py
+++ b/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:
diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index 79cf4526fa0c4..db4b421ef1f17 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1692,6 +1692,7 @@ def test_no_cell_ids(self):
         assert s.find('<td  class="data row0 col0" >') != -1
 
     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")]}]
@@ -1702,6 +1703,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")]}]})

From d7f7e7274989abd4f0e9d783c15a580683d35720 Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Sun, 20 Sep 2020 09:24:55 +0200
Subject: [PATCH 08/11] fix test after recent merge master

---
 pandas/tests/io/formats/test_style.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index 816f0106879b7..5dbb81d9e22fc 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1726,7 +1726,7 @@ def test_chaining_table_styles(self):
     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 = Styler(df, uuid_len=0)
         s = s.set_table_styles({"A": [{"selector": "", "props": [("color", "blue")]}]})
         assert "#T__ .col0 {\n          color: blue;\n    }" in s.render()
         s = s.set_table_styles(

From 0792752e2c0c61fd5dec5ee6138fcb7072052454 Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Thu, 5 Nov 2020 17:13:58 +0100
Subject: [PATCH 09/11] style fix

---
 pandas/tests/io/formats/test_style.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pandas/tests/io/formats/test_style.py b/pandas/tests/io/formats/test_style.py
index 95e584774ddca..64fe8a7730ae2 100644
--- a/pandas/tests/io/formats/test_style.py
+++ b/pandas/tests/io/formats/test_style.py
@@ -1714,7 +1714,7 @@ def test_set_data_classes(self, classes):
 
     def test_chaining_table_styles(self):
         # GH 35607
-        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
+        df = DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
         styler = df.style.set_table_styles(
             [{"selector": "", "props": [("background-color", "yellow")]}]
         ).set_table_styles(
@@ -1725,7 +1725,7 @@ def test_chaining_table_styles(self):
 
     def test_column_and_row_styling(self):
         # GH 35607
-        df = pd.DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
+        df = DataFrame(data=[[0, 1], [1, 2]], columns=["A", "B"])
         s = Styler(df, uuid_len=0)
         s = s.set_table_styles({"A": [{"selector": "", "props": [("color", "blue")]}]})
         assert "#T__ .col0 {\n          color: blue;\n    }" in s.render()

From 93d3f8dfeeb1a5c4d595825ada22596f1ba40296 Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Sun, 22 Nov 2020 10:11:51 +0100
Subject: [PATCH 10/11] TYPE: add type for set_table_styles

---
 pandas/io/formats/style.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index 638b4b1aed774..757f22b0db4c5 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -991,7 +991,15 @@ def set_caption(self, caption: str) -> "Styler":
         self.caption = caption
         return self
 
-    def set_table_styles(self, table_styles, axis=0, overwrite=True) -> "Styler":
+    def set_table_styles(
+        self,
+        table_styles: Union[
+            Dict[Union[str, Tuple], List[Dict[str, Union[str, List[Tuple[str, str]]]]]],
+            List[Dict[str, Union[str, List[Tuple[str, str]]]]],
+        ],
+        axis=0,
+        overwrite=True,
+    ) -> "Styler":
         """
         Set the table styles on a Styler.
 

From d5a6fbb420bda957e8298a11eaa003b552fa046c Mon Sep 17 00:00:00 2001
From: "JHM Darbyshire (iMac)" <attack68@users.noreply.github.com>
Date: Sun, 22 Nov 2020 15:25:00 +0100
Subject: [PATCH 11/11] TYPE: revert chgs

---
 pandas/io/formats/style.py | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index 757f22b0db4c5..638b4b1aed774 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -991,15 +991,7 @@ def set_caption(self, caption: str) -> "Styler":
         self.caption = caption
         return self
 
-    def set_table_styles(
-        self,
-        table_styles: Union[
-            Dict[Union[str, Tuple], List[Dict[str, Union[str, List[Tuple[str, str]]]]]],
-            List[Dict[str, Union[str, List[Tuple[str, str]]]]],
-        ],
-        axis=0,
-        overwrite=True,
-    ) -> "Styler":
+    def set_table_styles(self, table_styles, axis=0, overwrite=True) -> "Styler":
         """
         Set the table styles on a Styler.