From 81744a5c4df684cccd8c9bf1a31363325b7b2087 Mon Sep 17 00:00:00 2001 From: Pete Blois Date: Fri, 22 Sep 2023 14:40:13 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 567720645 --- google/colab/_reprs.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/google/colab/_reprs.py b/google/colab/_reprs.py index fe1d8951..f9810750 100644 --- a/google/colab/_reprs.py +++ b/google/colab/_reprs.py @@ -145,10 +145,22 @@ def _dataframe_intrinsic_repr(dataframe): } if ip := IPython.get_ipython(): namespace = ip.user_ns + found = False for varname, var in namespace.items(): if dataframe is var and not varname.startswith('_'): result['variable_name'] = varname + found = True break + if not found: + last_line = ip.user_ns['In'][-1] + varname, dot, operator = last_line.partition('.') + if varname.isidentifier() and dot and operator.startswith('head('): + import pandas as pd + + possible_df = ip.user_ns.get(varname) + if isinstance(possible_df, pd.DataFrame): + result['variable_name'] = varname + dataframe = possible_df if summary := _summarize_dataframe(dataframe): result['summary'] = summary