BUG: fix to_latex bold_rows option

Author: kdebrab

pandas/io/formats/format.py

@@ -943,6 +943,10 @@ def get_col_type(dtype):
                          if x else '{}') for x in row]
             else:
                 crow = [x if x else '{}' for x in row]
+            if self.fmt.kwds.get('bold_rows', True) and self.fmt.index:
+                # bold row labels
+                crow = ['\\textbf{%s}' % x if j < ilevels and x.strip() not in ['', '{}'] else x
+                        for j, x in enumerate(crow)]
             if i < clevels and self.fmt.header and self.multicolumn:
                 # sum up columns to multicolumns
                 crow = self._format_multicolumn(crow, ilevels)

pandas/tests/io/formats/test_to_latex.py

@@ -52,8 +52,8 @@ def test_to_latex(self, frame):
 \toprule
 {} &  a &   b \\
 \midrule
-0 &  1 &  b1 \\
-1 &  2 &  b2 \\
+\textbf{0} &  1 &  b1 \\
+\textbf{1} &  2 &  b2 \\
 \bottomrule
 \end{tabular}
 """
@@ -83,8 +83,8 @@ def test_to_latex_format(self, frame):
 \toprule
 {} &  a &   b \\
 \midrule
-0 &  1 &  b1 \\
-1 &  2 &  b2 \\
+\textbf{0} &  1 &  b1 \\
+\textbf{1} &  2 &  b2 \\
 \bottomrule
 \end{tabular}
 """
@@ -110,9 +110,9 @@ def test_to_latex_with_formatters(self):
 \toprule
 {} & datetime64 &  float & int &    object \\
 \midrule
-index: 0 &    2016-01 & [ 1.0] & 0x1 &  -(1, 2)- \\
-index: 1 &    2016-02 & [ 2.0] & 0x2 &    -True- \\
-index: 2 &    2016-03 & [ 3.0] & 0x3 &   -False- \\
+\textbf{index: 0} &    2016-01 & [ 1.0] & 0x1 &  -(1, 2)- \\
+\textbf{index: 1} &    2016-02 & [ 2.0] & 0x2 &    -True- \\
+\textbf{index: 2} &    2016-03 & [ 3.0] & 0x3 &   -False- \\
 \bottomrule
 \end{tabular}
 """
@@ -126,7 +126,7 @@ def test_to_latex_multiindex(self):
 {} &  x \\
 {} &  y \\
 \midrule
-0 &  a \\
+\textbf{0} &  a \\
 \bottomrule
 \end{tabular}
 """
@@ -138,7 +138,7 @@ def test_to_latex_multiindex(self):
 \toprule
   &   &  0 \\
 \midrule
-x & y &  a \\
+\textbf{x} & \textbf{y} &  a \\
 \bottomrule
 \end{tabular}
 """
@@ -157,11 +157,11 @@ def test_to_latex_multiindex(self):
 \toprule
    &   &  0 &  1 &  2 &  3 \\
 \midrule
-c1 & 0 &  0 &  1 &  2 &  3 \\
-   & 1 &  4 &  5 &  6 &  7 \\
-c2 & 0 &  0 &  1 &  2 &  3 \\
-   & 1 &  4 &  5 &  6 &  7 \\
-c3 & 0 &  0 &  1 &  2 &  3 \\
+\textbf{c1} & \textbf{0} &  0 &  1 &  2 &  3 \\
+   & \textbf{1} &  4 &  5 &  6 &  7 \\
+\textbf{c2} & \textbf{0} &  0 &  1 &  2 &  3 \\
+   & \textbf{1} &  4 &  5 &  6 &  7 \\
+\textbf{c3} & \textbf{0} &  0 &  1 &  2 &  3 \\
 \bottomrule
 \end{tabular}
 """
@@ -174,13 +174,13 @@ def test_to_latex_multiindex(self):
         result = df.to_latex()
         expected = r"""\begin{tabular}{lrrrrr}
 \toprule
-a & \multicolumn{2}{l}{c1} & \multicolumn{2}{l}{c2} & c3 \\
-b &  0 &  1 &  0 &  1 &  0 \\
+\textbf{a} & \multicolumn{2}{l}{c1} & \multicolumn{2}{l}{c2} & c3 \\
+\textbf{b} &  0 &  1 &  0 &  1 &  0 \\
 \midrule
-0 &  0 &  4 &  0 &  4 &  0 \\
-1 &  1 &  5 &  1 &  5 &  1 \\
-2 &  2 &  6 &  2 &  6 &  2 \\
-3 &  3 &  7 &  3 &  7 &  3 \\
+\textbf{0} &  0 &  4 &  0 &  4 &  0 \\
+\textbf{1} &  1 &  5 &  1 &  5 &  1 \\
+\textbf{2} &  2 &  6 &  2 &  6 &  2 \\
+\textbf{3} &  3 &  7 &  3 &  7 &  3 \\
 \bottomrule
 \end{tabular}
 """
@@ -194,12 +194,12 @@ def test_to_latex_multiindex(self):
         expected = r"""\begin{tabular}{llr}
 \toprule
   &   &  c \\
-a & b &    \\
+\textbf{a} & \textbf{b} &    \\
 \midrule
-0 & a &  1 \\
-  & b &  2 \\
-1 & a &  3 \\
-  & b &  4 \\
+\textbf{0} & \textbf{a} &  1 \\
+  & \textbf{b} &  2 \\
+\textbf{1} & \textbf{a} &  3 \\
+  & \textbf{b} &  4 \\
 \bottomrule
 \end{tabular}
 """
@@ -211,10 +211,10 @@ def test_to_latex_multiindex(self):
 \toprule
 {} & \multicolumn{8}{l}{c} \\
 {} & count & mean &       std &  min &   25\% &  50\% &   75\% &  max \\
-a &       &      &           &      &       &      &       &      \\
+\textbf{a} &       &      &           &      &       &      &       &      \\
 \midrule
-0 &   2.0 &  1.5 &  0.707107 &  1.0 &  1.25 &  1.5 &  1.75 &  2.0 \\
-1 &   2.0 &  3.5 &  0.707107 &  3.0 &  3.25 &  3.5 &  3.75 &  4.0 \\
+\textbf{0} &   2.0 &  1.5 &  0.707107 &  1.0 &  1.25 &  1.5 &  1.75 &  2.0 \\
+\textbf{1} &   2.0 &  3.5 &  0.707107 &  3.0 &  3.25 &  3.5 &  3.75 &  4.0 \\
 \bottomrule
 \end{tabular}
 """
@@ -235,11 +235,11 @@ def test_to_latex_multicolumnrow(self):
 {} & \multicolumn{2}{l}{c1} & \multicolumn{2}{l}{c2} & c3 \\
 {} &  0 &  1 &  0 &  1 &  0 \\
 \midrule
-0 &  0 &  5 &  0 &  5 &  0 \\
-1 &  1 &  6 &  1 &  6 &  1 \\
-2 &  2 &  7 &  2 &  7 &  2 \\
-3 &  3 &  8 &  3 &  8 &  3 \\
-4 &  4 &  9 &  4 &  9 &  4 \\
+\textbf{0} &  0 &  5 &  0 &  5 &  0 \\
+\textbf{1} &  1 &  6 &  1 &  6 &  1 \\
+\textbf{2} &  2 &  7 &  2 &  7 &  2 \\
+\textbf{3} &  3 &  8 &  3 &  8 &  3 \\
+\textbf{4} &  4 &  9 &  4 &  9 &  4 \\
 \bottomrule
 \end{tabular}
 """
@@ -251,11 +251,11 @@ def test_to_latex_multicolumnrow(self):
 {} & c1 &    & c2 &    & c3 \\
 {} &  0 &  1 &  0 &  1 &  0 \\
 \midrule
-0 &  0 &  5 &  0 &  5 &  0 \\
-1 &  1 &  6 &  1 &  6 &  1 \\
-2 &  2 &  7 &  2 &  7 &  2 \\
-3 &  3 &  8 &  3 &  8 &  3 \\
-4 &  4 &  9 &  4 &  9 &  4 \\
+\textbf{0} &  0 &  5 &  0 &  5 &  0 \\
+\textbf{1} &  1 &  6 &  1 &  6 &  1 \\
+\textbf{2} &  2 &  7 &  2 &  7 &  2 \\
+\textbf{3} &  3 &  8 &  3 &  8 &  3 \\
+\textbf{4} &  4 &  9 &  4 &  9 &  4 \\
 \bottomrule
 \end{tabular}
 """
@@ -266,13 +266,13 @@ def test_to_latex_multicolumnrow(self):
 \toprule
    &   &  0 &  1 &  2 &  3 &  4 \\
 \midrule
-\multirow{2}{*}{c1} & 0 &  0 &  1 &  2 &  3 &  4 \\
-   & 1 &  5 &  6 &  7 &  8 &  9 \\
+\multirow{2}{*}{\textbf{c1}} & \textbf{0} &  0 &  1 &  2 &  3 &  4 \\
+   & \textbf{1} &  5 &  6 &  7 &  8 &  9 \\
 \cline{1-7}
-\multirow{2}{*}{c2} & 0 &  0 &  1 &  2 &  3 &  4 \\
-   & 1 &  5 &  6 &  7 &  8 &  9 \\
+\multirow{2}{*}{\textbf{c2}} & \textbf{0} &  0 &  1 &  2 &  3 &  4 \\
+   & \textbf{1} &  5 &  6 &  7 &  8 &  9 \\
 \cline{1-7}
-c3 & 0 &  0 &  1 &  2 &  3 &  4 \\
+\textbf{c3} & \textbf{0} &  0 &  1 &  2 &  3 &  4 \\
 \bottomrule
 \end{tabular}
 """
@@ -286,13 +286,13 @@ def test_to_latex_multicolumnrow(self):
    &   & \multicolumn{2}{c}{c1} & \multicolumn{2}{c}{c2} & c3 \\
    &   &  0 &  1 &  0 &  1 &  0 \\
 \midrule
-\multirow{2}{*}{c1} & 0 &  0 &  1 &  2 &  3 &  4 \\
-   & 1 &  5 &  6 &  7 &  8 &  9 \\
+\multirow{2}{*}{\textbf{c1}} & \textbf{0} &  0 &  1 &  2 &  3 &  4 \\
+   & \textbf{1} &  5 &  6 &  7 &  8 &  9 \\
 \cline{1-7}
-\multirow{2}{*}{c2} & 0 &  0 &  1 &  2 &  3 &  4 \\
-   & 1 &  5 &  6 &  7 &  8 &  9 \\
+\multirow{2}{*}{\textbf{c2}} & \textbf{0} &  0 &  1 &  2 &  3 &  4 \\
+   & \textbf{1} &  5 &  6 &  7 &  8 &  9 \\
 \cline{1-7}
-c3 & 0 &  0 &  1 &  2 &  3 &  4 \\
+\textbf{c3} & \textbf{0} &  0 &  1 &  2 &  3 &  4 \\
 \bottomrule
 \end{tabular}
 """
@@ -315,8 +315,8 @@ def test_to_latex_escape(self):
 \toprule
 {} & co$e^x$ & co^l1 \\
 \midrule
-a &       a &     a \\
-b &       b &     b \\
+\textbf{a} &       a &     a \\
+\textbf{b} &       b &     b \\
 \bottomrule
 \end{tabular}
 '''
@@ -325,8 +325,8 @@ def test_to_latex_escape(self):
 \toprule
 {} & co\$e\textasciicircumx\$ & co\textasciicircuml1 \\
 \midrule
-a &       a &     a \\
-b &       b &     b \\
+\textbf{a} &       a &     a \\
+\textbf{b} &       b &     b \\
 \bottomrule
 \end{tabular}
 '''
@@ -351,8 +351,8 @@ def test_to_latex_longtable(self, frame):
 
 \bottomrule
 \endlastfoot
-0 &  1 &  b1 \\
-1 &  2 &  b2 \\
+\textbf{0} &  1 &  b1 \\
+\textbf{1} &  2 &  b2 \\
 \end{longtable}
 """
 
@@ -387,16 +387,16 @@ def test_to_latex_escape_special_chars(self):
 \toprule
 {} &  0 \\
 \midrule
-0 &  \& \\
-1 &  \% \\
-2 &  \$ \\
-3 &  \# \\
-4 &  \_ \\
-5 &  \{ \\
-6 &  \} \\
-7 &  \textasciitilde \\
-8 &  \textasciicircum \\
-9 &  \textbackslash \\
+\textbf{0} &  \& \\
+\textbf{1} &  \% \\
+\textbf{2} &  \$ \\
+\textbf{3} &  \# \\
+\textbf{4} &  \_ \\
+\textbf{5} &  \{ \\
+\textbf{6} &  \} \\
+\textbf{7} &  \textasciitilde \\
+\textbf{8} &  \textasciicircum \\
+\textbf{9} &  \textbackslash \\
 \bottomrule
 \end{tabular}
 """
@@ -409,8 +409,8 @@ def test_to_latex_no_header(self):
         withindex_result = df.to_latex(header=False)
         withindex_expected = r"""\begin{tabular}{lrl}
 \toprule
-0 &  1 &  b1 \\
-1 &  2 &  b2 \\
+\textbf{0} &  1 &  b1 \\
+\textbf{1} &  2 &  b2 \\
 \bottomrule
 \end{tabular}
 """
@@ -436,8 +436,8 @@ def test_to_latex_specified_header(self):
 \toprule
 {} & AA &  BB \\
 \midrule
-0 &  1 &  b1 \\
-1 &  2 &  b2 \\
+\textbf{0} &  1 &  b1 \\
+\textbf{1} &  2 &  b2 \\
 \bottomrule
 \end{tabular}
 """
@@ -462,8 +462,8 @@ def test_to_latex_specified_header(self):
 \toprule
 {} & $A$ & $B$ \\
 \midrule
-0 &   1 &  b1 \\
-1 &   2 &  b2 \\
+\textbf{0} &   1 &  b1 \\
+\textbf{1} &   2 &  b2 \\
 \bottomrule
 \end{tabular}
 """
@@ -484,8 +484,8 @@ def test_to_latex_decimal(self, frame):
 \toprule
 {} &    a &   b \\
 \midrule
-0 &  1,0 &  b1 \\
-1 &  2,1 &  b2 \\
+\textbf{0} &  1,0 &  b1 \\
+\textbf{1} &  2,1 &  b2 \\
 \bottomrule
 \end{tabular}
 """
@@ -499,10 +499,40 @@ def test_to_latex_series(self):
 \toprule
 {} &  0 \\
 \midrule
-0 &  a \\
-1 &  b \\
-2 &  c \\
+\textbf{0} &  a \\
+\textbf{1} &  b \\
+\textbf{2} &  c \\
 \bottomrule
 \end{tabular}
 """
         assert withindex_result == withindex_expected
+
+    def test_to_latex_bold_rows(self):
+        # GH 16707
+        df = DataFrame({'a': [1, 2], 'b': ['b1', 'b2']})
+        observed = df.to_latex(bold_rows=True)
+        expected = r"""\begin{tabular}{lrl}
+\toprule
+\textbf{} &  a &   b \\
+\midrule
+\textbf{0} &  1 &  b1 \\
+\textbf{1} &  2 &  b2 \\
+\bottomrule
+\end{tabular}
+"""
+        assert observed == expected
+
+    def test_to_latex_no_bold_rows(self):
+        # GH 16707
+        df = DataFrame({'a': [1, 2], 'b': ['b1', 'b2']})
+        observed = df.to_latex(bold_rows=False)
+        expected = r"""\begin{tabular}{lrl}
+\toprule
+{} &  a &   b \\
+\midrule
+0 &  1 &  b1 \\
+1 &  2 &  b2 \\
+\bottomrule
+\end{tabular}
+"""
+        assert observed == expected