ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 07:35:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Pandas with multi-index (issue #399)

Browse Source
This commit is contained in:
allegroai 2021-07-23 15:57:38 +03:00
parent 08da22296a
commit 900ee9de5a
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/utilities/plotly_reporter.py
View File

@ -503,6 +503,10 @@ def create_plotly_table(table_plot, title, series, layout_config=None):
headers_values = list([col] for col in table_plot.columns) headers_values = list([col] for col in table_plot.columns)
cells_values = table_plot.T.values.tolist() cells_values = table_plot.T.values.tolist()
if index_added: if index_added:
if isinstance(table_plot.index, pd.MultiIndex):
headers_values = list(table_plot.index.names or []) + headers_values
cells_values = list(zip(*(table_plot.index.values.tolist() or []))) + cells_values
else:
headers_values.insert(0, table_plot.index.name or "") headers_values.insert(0, table_plot.index.name or "")
cells_values.insert(0, table_plot.index.values.tolist()) cells_values.insert(0, table_plot.index.values.tolist())