loop through two dataframes and plot by column in loop

Question

I have 2 dataframes with the same column names, index and size. I want to create a scatterplot of one dataframe column vs the other with the same header. When I try the code below, only the sim.columns are looping, while the obs.columns plots only the first column and don't loop. So what I get are scatter plots of each sim.columns against the first obs.column only. I'm not sure what is messed up with this loop. Thanks for your help!

    obs= pd.read_csv(obsFile)
    obs.rename(columns={obs.columns[0]: "SP" }, inplace = True)
    sim= pd.read_csv(simFile)
    sim.rename(columns={sim.columns[0]: "SP" }, inplace = True)
    
sim = sim.set_index("SP")
obs = obs.set_index("SP")

for colsim in sim.columns:
    for colobs in obs.columns:
        axes = plt.gca()
        axes.set_xlim([1,630])
        plt.scatter(sim.index, sim[colsim])
        plt.scatter(obs.index, obs[colobs])

        plt.xlabel('Stress Period')
        plt.ylabel('groundwater elevation(m)')
        plt.title(str(colsim))
        plt.savefig(os.path.join(outFold, str(colsim)+'.pdf')) 
        plt.close()
        break

U13-Forward · Accepted Answer · 2021-07-16 01:30:06Z

1

Instead of a double for loop:

for colsim in sim.columns:
    for colobs in obs.columns:
        ...

Just use a zip:

for colsim, colobs in zip(sim.columns, obs.columns):
    ...

Not sure what you mean with what you wanna do, but if this code doesn't work, try removing the break then it should work, but if it works without removing the break just keep it.

answered Jul 16, 2021 at 1:30

U13-Forward

71.8k15 gold badges100 silver badges125 bronze badges

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

1 Comment

BENY Over a year ago

zip is different from double loop ~

ffm_nosoup · Accepted Answer · 2021-07-16 01:30:00Z

0

You shouldn't use a nested loop here, it does not what you want to achieve. Instead you want to loop through both dataframes at the same time in one loop. i can think of 2 ways how to achieve this:

merge the dataframes and loop through the merged dataframe in one loop
introduce a counter variable which can be used as an index for looping through both dataframes parallel

answered Jul 16, 2021 at 1:30

ffm_nosoup

1075 bronze badges

2 Comments

U13-Forward Over a year ago

Just using zip is enough... Like in my answer

ffm_nosoup Over a year ago

you are right, i used the wrong term. by merging i meant zipping.

Collectives™ on Stack Overflow

loop through two dataframes and plot by column in loop

2 Answers 2

1 Comment

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related