string or categorical integer values considered as numeric in strip/box/violin plots
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 18.8k
- Forks
- 2.8k
- Avg merge
- 16h 26m
- Merged PRs (30d)
- 21
Description
Issue:
When using integer values as categorical variable in a strip / box / violin plot, the values of the categorical variable are mapped to a continuous numeric axis even if the values are of string or pd.Categorical type.
Example:
We create a dataframe with columns names having an integer value in string format. These could be any category that makes sense to the specific business case (e.g. product code, etc.)
import numpy as np
import pandas as pd
import plotly_express as px
n = 50
df = pd.DataFrame({
'1': np.random.normal(2, .3, n),
'2': np.random.lognormal(.5, .2, n),
'34': np.random.triangular(0, 2, 3, n),
'123': np.random.uniform(1, 3, n)
})
We unpivot the data using and make a strip plot.
df1 = df.melt()
px.strip(df1, x='variable', y='value')

The categorical variable (values '1', '2', '34' and '123') get mapped to a continuous numeric scale. Here, the variables '1' and '2' blend together and this can get worst if there are orders of magnitude between the different values.
Converting the string values to pd.Categorical type yields the same result as above.
df2 = df.melt()
df2.variable = pd.Categorical(df2.variable)
px.strip(df2, x='variable', y='value')
Workaround:
Adding a character to the values makes them be recognized as categorical which is the expected result (except for the added character in the category names). Unfortunately, adding a blank space does not work either.
df3 = df.copy()
df3.columns = [f"c{c}" for c in df3.columns]
px.strip(df3.melt(), x='variable', y='value')

Considering that numeric categorical values are legitimate in many contexts, it should be possible to use numbers as categories if they are represented by a string or categorical data type, as is the case with the color parameter (https://github.com/plotly/plotly_express/issues/140):
px.strip(df.melt(), y='value', color='variable')

Thanks!
Package Version
-------------------- ---------
plotly 4.1.1
plotly-express 0.4.1
@emmanuelle, this is one of the two issues we discussed at the PyData meetup.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the px.strip(df.melt(), x='variable', y='value') reproducer and compare it with the color='variable' example. Trace how categorical values are inferred for strip, box, and violin plots; done means string and pd.Categorical labels containing integer-like values remain discrete categories without a prefix. No file or test location is named.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- numpy, pandas, python
- Domain
- data-visualization
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100