ENH: changes requested in issue

Author: attack68

doc/source/whatsnew/v1.1.1.rst

@@ -72,11 +72,6 @@ Categorical
 - 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:

doc/source/whatsnew/v1.2.0.rst

@@ -51,6 +51,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`)
 -
 -
 

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

@@ -1689,6 +1689,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")]}]
@@ -1699,6 +1700,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")]}]})