code: def remove_col_suffixes(df):
      df.columns = [w.split("_")[0] for w in df.columns]
      return df
  
file_path: pyLDAvis/tests/pyLDAvis/test_prepare.py
imports: import json
  import os.path
  import funcy
  from numpy.testing import assert_array_equal
  import numpy
  import pandas
  from pandas.testing import assert_frame_equal
  from pyLDAvis import prepare
label_desc: Use image node_id 2 for calling a global function with example usage: remove_col_suffixes(df) and returns: df