code: def highlight_max(data, color="yellow"):
      """highlight the maximum in a Series or DataFrame"""
      attr = "background-color: {}".format(color)
      if data.ndim == 1:  # Series from .apply(axis=0) or axis=1
          is_max = data == data.max()
          return [attr if v else "" for v in is_max]
      else:  # from .apply(axis=None)
          is_max = data == data.max().max()
          return pd.DataFrame(
              np.where(is_max, attr, ""),
              index=data.index,
              columns=data.columns,
          )
  
imports: import numpy
  import pandas
  import streamlit
label_desc: Use image node_id 2 for calling a global function with example usage: highlight_max(data, color) and returns: unknown, pd