# Buggy code
def apply(self, func, axis=0, subset=None, **kwargs):
    """
    Apply a function column-wise, row-wise, or table-wase,
    updating the HTML representation with the result.

    .. versionadded:: 0.17.1

    Parameters
    ----------
    func: function
    axis: int, str or None
        apply to each column (``axis=0`` or ``'index'``)
        or to each row (``axis=1`` or ``'columns'``) or
        to the entire DataFrame at once with ``axis=None``.
    subset: IndexSlice
        a valid indexer to limit ``data`` to *before* applying the
        function. Consider using a pandas.IndexSlice
    kwargs: dict
        pass along to ``func``

    Returns
    -------
    self

    Notes
    -----
    This is similar to ``DataFrame.apply``, except that ``axis=None``
    applies the function to the entire DataFrame at once,
    rather than column-wise or row-wise.
    """
    self._todo.append(
        (lambda instance: getattr(instance, "_apply"), (func, axis, subset), kwargs)
    )
    return self


# Traceback (KeyError)
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
/usr/local/lib/python3.5/site-packages/IPython/core/formatters.py in __call__(self, obj)
339             method = _safe_get_formatter_method(obj, self.print_method)
340             if method is not None:
--> 341                 return method()
342             return None
343         else:

/usr/local/lib/python3.5/site-packages/pandas/core/style.py in _repr_html_(self)
158         Hooks into Jupyter notebook rich display system.
159         '''
--> 160         return self.render()
161
162     def _translate(self):

/usr/local/lib/python3.5/site-packages/pandas/core/style.py in render(self)
259         """
260         self._compute()
--> 261         d = self._translate()
262         # filter out empty styles, every cell will have a class
263         # but the list of props may just be [['', '']].

/usr/local/lib/python3.5/site-packages/pandas/core/style.py in _translate(self)
220                 cs = [DATA_CLASS, "row%s" % r, "col%s" % c]
221                 cs.extend(cell_context.get("data", {}).get(r, {}).get(c, []))
--> 222                 row_es.append({"type": "td", "value": self.data.iloc[r][c],
223                                "class": " ".join(cs), "id": "_".join(cs[1:])})
224                 props = []

/usr/local/lib/python3.5/site-packages/pandas/core/series.py in __getitem__(self, key)
555     def __getitem__(self, key):
556         try:
--> 557             result = self.index.get_value(self, key)
558
559             if not np.isscalar(result):

/usr/local/lib/python3.5/site-packages/pandas/core/index.py in get_value(self, series, key)
1788
1789         try:
-> 1790             return self._engine.get_value(s, k)
1791         except KeyError as e1:
1792             if len(self) > 0 and self.inferred_type in ['integer','boolean']:

pandas/index.pyx in pandas.index.IndexEngine.get_value (pandas/index.c:3204)()

pandas/index.pyx in pandas.index.IndexEngine.get_value (pandas/index.c:2903)()

pandas/index.pyx in pandas.index.IndexEngine.get_loc (pandas/index.c:3843)()

pandas/hashtable.pyx in pandas.hashtable.Int64HashTable.get_item (pandas/hashtable.c:6525)()

pandas/hashtable.pyx in pandas.hashtable.Int64HashTable.get_item (pandas/hashtable.c:6463)()

KeyError: 0

# Fixed code
def apply(self, func, axis=0, subset=None, **kwargs):
    """
    Apply a function column-wise, row-wise, or table-wase,
    updating the HTML representation with the result.

    .. versionadded:: 0.17.1

    Parameters
    ----------
    func: function
    axis: int, str or None
        apply to each column (``axis=0`` or ``'index'``)
        or to each row (``axis=1`` or ``'columns'``) or
        to the entire DataFrame at once with ``axis=None``.
    subset: IndexSlice
        a valid indexer to limit ``data`` to *before* applying the
        function. Consider using a pandas.IndexSlice
    kwargs: dict
        pass along to ``func``

    Returns
    -------
    self : Styler

    Notes
    -----
    This is similar to ``DataFrame.apply``, except that ``axis=None``
    applies the function to the entire DataFrame at once,
    rather than column-wise or row-wise.
    """
    self._todo.append(
        (lambda instance: getattr(instance, "_apply"), (func, axis, subset), kwargs)
    )
    return self