Change the color of lines with categorical column values

Asked Oct 05 '21 at 16:33

Active Oct 05 '21 at 17:45

Viewed 27 times

I have the following code:

df1['salary'].groupby(df1['job']).mean().plot(kind='line')

This gives one line as output. I want to have two lines in the output based on another column which is gender.

I tried the following code, but it did not work.

df1['salary'].groupby(df1['job']).mean().plot(kind='line', color=df1['gender'])

I would appreciate if anyone can help me.

edited Oct 05 '21 at 17:45

Trenton McKinney

56,955
33
144
158

asked Oct 05 '21 at 16:33

user4172070

This is a typical case for which seaborn has been invented: `sns.lineplot(data=df1, x='job', y='salary', hue='gender')`. – JohanC Oct 05 '21 at 17:06
@JohanC Thank you so much. I have been searching for 4 hours just for this code! I appreciate it. – user4172070 Oct 05 '21 at 17:42
https://stackoverflow.com/q/47104862/7758804, https://stackoverflow.com/q/32525718/7758804 – Trenton McKinney Oct 05 '21 at 17:47
@TrentonMcKinney I am wondering when we use y in this code: sns.lineplot(data=df1, x='job', y='salary', hue='gender') does it consider average salary by default?? – user4172070 Oct 05 '21 at 17:50
If there are values to be aggregated, that what `estimator` : _name of pandas method or callable or None_ is for (e.g. `estimator='mean'`), which is the default. – Trenton McKinney Oct 05 '21 at 17:51
@TrentonMcKinney thank you so much. what if I want to count values. df1['job'].value_counts().plot(kind='line') and change color by gender – user4172070 Oct 05 '21 at 17:54
Use [`seaborn.countplot`](https://seaborn.pydata.org/generated/seaborn.countplot.html). Otherwise, use `estimator='size'` or maybe `estimator='count'`. – Trenton McKinney Oct 05 '21 at 17:55
@TrentonMcKinney but I want it to be line. countplot gives bar chart. how should i change it to line chart? – user4172070 Oct 05 '21 at 17:59
Something like `pd.crosstab(df1.salary, df1.job).plot()`. It's really a different question. – Trenton McKinney Oct 05 '21 at 18:08
1

Thank you so much :) – user4172070 Oct 05 '21 at 18:10

Change the color of lines with categorical column values

0 Answers0