Commit cc8a1a1

committed

fix: custom category order was hard-coded

1 parent 5e079e1 commit cc8a1a1Copy full SHA for cc8a1a1

File tree

+27

-2

lines changed

packages/python/plotly/plotly
- express
  - _core.py
- tests/test_optional/test_px
  - test_px_functions.py

+27

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -2159,11 +2159,11 @@ def process_dataframe_pie(args, trace_patch):`
`2159`	`2159`	`uniques = df.get_column(names).unique(maintain_order=True).to_list()`
`2160`	`2160`	`order = [x for x in OrderedDict.fromkeys(list(order_in) + uniques) if x in uniques]`
`2161`	`2161`
`2162`		- # Sort args['data_frame'] by column 'b' according to order `order`.
	`2162`	+ # Sort args['data_frame'] by column `names` according to order `order`.
`2163`	`2163`	`token = nw.generate_temporary_column_name(8, df.columns)`
`2164`	`2164`	`args["data_frame"] = (`
`2165`	`2165`	`df.with_columns(`
`2166`		`- nw.col("b")`
	`2166`	`+ nw.col(names)`
`2167`	`2167`	`.replace_strict(order, range(len(order)), return_dtype=nw.UInt32)`
`2168`	`2168`	`.alias(token)`
`2169`	`2169`	`)`

Lines changed: 25 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,31 @@ def test_pie_like_px():`
`56`	`56`	`_compare_figures(trace, fig)`
`57`	`57`
`58`	`58`
	`59`	`+def test_pie_custom_category_order(constructor):`
	`60`	`+ # https://github.com/plotly/plotly.py/issues/4999`
	`61`	`+ df = constructor(`
	`62`	`+ {`
	`63`	`+ "status": ["On Route", "Pending", "Waiting Result", "Delivered"],`
	`64`	`+ "count": [28, 10, 73, 8],`
	`65`	`+ }`
	`66`	`+ )`
	`67`	`+ custom_order = ["Pending", "Waiting Result", "On Route", "Delivered"]`
	`68`	`+ result = px.pie(`
	`69`	`+ data_frame=df,`
	`70`	`+ values="count",`
	`71`	`+ names="status",`
	`72`	`+ category_orders={`
	`73`	`+ "status": custom_order`
	`74`	`+ }, # raises polars.exceptions.ColumnNotFoundError`
	`75`	`+ ).to_dict()`
	`76`	`+ assert list(result["data"][0]["labels"]) == [`
	`77`	`+ "Pending",`
	`78`	`+ "Waiting Result",`
	`79`	`+ "On Route",`
	`80`	`+ "Delivered",`
	`81`	`+ ]`
	`82`	`+`
	`83`	`+`
`59`	`84`	`def test_sunburst_treemap_colorscales():`
`60`	`85`	`labels = ["Eve", "Cain", "Seth", "Enos", "Noam", "Abel", "Awan", "Enoch", "Azura"]`
`61`	`86`	`parents = ["", "Eve", "Eve", "Seth", "Seth", "Eve", "Eve", "Awan", "Eve"]`

Comments

(0)