stahsing

Author: TomAugspurger

pandas/core/frame.py

@@ -1393,6 +1393,56 @@ def to_html(self, buf=None, columns=None, col_space=None, colSpace=None,
         if buf is None:
             return formatter.buf.getvalue()
 
+    def style(self, rows=None, columns=None, frame=None):
+        row_styles, col_styles, frame_styles = self._build_styles(
+            rows=rows, columns=columns, frame=frame
+        )
+        return self.to_html(styles)
+
+    def _format_sytle_args(self, rows=None, columns=None, frame=None):
+        rows = {row: [x] if not com.is_list_like(x) else x
+                for row, x in rows.items()}
+        columns = {col: [x] if not com.is_list_like(x) else x
+                for col, x in columns.items()}
+        frame = [x] if not com.is_list_like(x) else x
+        return rows, columns, frame
+
+
+    def _build_styles(self, rows=None, columns=None, frame=None):
+        final = []  # (row, col, prop, value)
+
+        rows, columns, frame = self._format_sytle_args(rows, columns, frame)
+
+        if rows is not None:
+            for row in rows:
+                for func in rows[row]:
+                    props, values  = zip(*func(self.loc[row]))
+                    final.append(pd.DataFrame({'row': row, 'col': self.columns,
+                                               'prop': props, 'value': values}))
+        if columns is not None:
+            for col in columns:
+                for func in columns[col]:
+                    props, values  = zip(*func(self[col]))
+                    final.append(pd.DataFrame({'row': self.index, 'col': col,
+                                               'prop': props, 'value': values}))
+
+        if frame is not None:
+            for func in frame:
+                r = func(self)
+                for col in r:
+                    try:
+                        prop, values = zip(*r[col])
+                    except ValueError:
+                        prop, values = '', ''
+                        final.append(pd.DataFrame({'row': r.index, 'col': col,
+                                               'prop': props, 'value': values}))
+
+        r = pd.concat(final)[['row', 'col', 'prop', 'value']]
+        r = (r['prop'] + ':' + r['value']).groupby([r['row'], r['col']]).agg(
+            lambda x: '; '.join(x))
+
+        return r.unstack()
+
     @Appender(fmt.docstring_to_string, indents=1)
     def to_latex(self, buf=None, columns=None, col_space=None, colSpace=None,
                  header=True, index=True, na_rep='NaN', formatters=None,