Skip to content

Commit e2886e8

Browse files
authored
Merge pull request plotly#2163 from plotly/pandas-1.0
fixed pandas 1.0 problem in sunburst/treemap example
2 parents d6d6aa3 + ecded6b commit e2886e8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

doc/python/sunburst-charts.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ def build_hierarchical_dataframe(df, levels, value_column, color_columns=None):
326326
df_all_trees = pd.DataFrame(columns=['id', 'parent', 'value', 'color'])
327327
for i, level in enumerate(levels):
328328
df_tree = pd.DataFrame(columns=['id', 'parent', 'value', 'color'])
329-
dfg = df.groupby(levels[i:]).sum(numerical_only=True)
329+
dfg = df.groupby(levels[i:]).sum()
330330
dfg = dfg.reset_index()
331331
df_tree['id'] = dfg[level].copy()
332332
if i < len(levels) - 1:

doc/python/treemaps.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ def build_hierarchical_dataframe(df, levels, value_column, color_columns=None):
278278
df_all_trees = pd.DataFrame(columns=['id', 'parent', 'value', 'color'])
279279
for i, level in enumerate(levels):
280280
df_tree = pd.DataFrame(columns=['id', 'parent', 'value', 'color'])
281-
dfg = df.groupby(levels[i:]).sum(numerical_only=True)
281+
dfg = df.groupby(levels[i:]).sum()
282282
dfg = dfg.reset_index()
283283
df_tree['id'] = dfg[level].copy()
284284
if i < len(levels) - 1:

0 commit comments

Comments
 (0)