s.k :
Question
How to shade or colorize the background of a seaborn plot using a column of a dataframe?
Code snippet
import numpy as np
import seaborn as sns; sns.set()
import matplotlib.pyplot as plt
fmri = sns.load_dataset("fmri")
fmri.sort_values('timepoint',inplace=True)
ax = sns.lineplot(x="timepoint", y="signal", data=fmri)
arr = np.ones(len(fmri))
arr[:300] = 0
arr[600:] = 2
fmri['background'] = arr
ax = sns.lineplot(x="timepoint", y="signal", hue="event", data=fmri)
Which produced this graph:
Desired output
What I'd like to have, according to the value in the new column 'background' and any palette or user defined colors, something like this:
jjsantoso :
ax.axvspan() could work for you, assuming backgrounds don't overlap over timepoints.
import numpy as np
import seaborn as sns; sns.set()
import matplotlib.pyplot as plt
fmri = sns.load_dataset("fmri")
fmri.sort_values('timepoint',inplace=True)
arr = np.ones(len(fmri))
arr[:300] = 0
arr[600:] = 2
fmri['background'] = arr
fmri['background'] = fmri['background'].astype(int).astype(str).map(lambda x: 'C'+x)
ax = sns.lineplot(x="timepoint", y="signal", hue="event", data=fmri)
ranges = fmri.groupby('background')['timepoint'].agg(['min', 'max'])
for i, row in ranges.iterrows():
ax.axvspan(xmin=row['min'], xmax=row['max'], facecolor=i, alpha=0.3)
Guess you like
Origin http://10.200.1.11:23101/article/api/json?id=379182&siteId=1