Using header of different dataframe as title in for loop in pandas

Question

I managed to plot a few charts out with using this for loop:

for i in df.columns[2:7]:
   df.plot.scatter(x='out_date', y=i, figsize = (10,7))
   plt.axvline(x=cutoff_date, color='r')
   plt.xlabel('out date')
   plt.ylabel('sucesses')

I wanted to add titles to the plots using the header of a different dataframe with the following code but the title will only be added to the last plot instead of every plot:

for x in df2.columns[67:72]:
   plt.title(x)

Is there a way to fix this?

Corralien · Accepted Answer · 2022-02-09 07:13:01Z

1

Try to zip your 2 dataframes:

for i, title in zip(df.columns[2:7], df2.columns[67:72]):
   ax = df.plot.scatter(x='out_date', y=i, figsize = (10,7), title=title)
   ax.axvline(x=cutoff_date, color='r')
   ax.set_xlabel('out date')
   ax.set_ylabel('sucesses')

You can also use AxesSubplot instance returned by df.plot methods instead of functions of plt module.

edited Feb 9, 2022 at 7:13

answered Feb 9, 2022 at 7:07

Corralien

121k8 gold badges43 silver badges68 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Using header of different dataframe as title in for loop in pandas

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related