Seaborn split column names for hue

Asked Jun 14 '22 at 16:48

Active Jun 14 '22 at 17:05

Viewed 368 times

I have a dataframe like this.

df = pd.DataFrame(np.array([[1, 2, 3, 4], [4, 5, 6, 4], [7, 8, 9, 4]]),
                   columns=['NP_A', 'NP_B', 'NP_C', "OP_A"])

I would like to use sns.lineplot.

But instead of the name of the columns as 'hue' I would like to make a split of the names.

"NP", "A"
"NP", "B"
...

In order to have a legend that considers the two combinations.

asked Jun 14 '22 at 16:48

Simone Panico

1. Convert `df` to long form (which you'll need for plotting anyway): `dfm = df.melt()` 2. Split the string into a new column: `dfm['hues'] = dfm.variable.str.split('_', expand=True)[1]` – Trenton McKinney Jun 14 '22 at 17:00
`dfm[['variable', 'hues']] = dfm.variable.str.split('_', expand=True)` – Trenton McKinney Jun 14 '22 at 17:09

0 Answers0