How do i iterate my dataframe values to create subplots?

Question

Hello all im trying to create a 11X2 subplot by iterating the columns that i have.

here is a snapshot of my dataframe. There are n units (100 actually) where each units have i amount of cycles.

The regression of sensor S7 for every unit combined looks like this:

, it is achieved by this:

for i in range(1,101):
    plt.plot(df[df.unit==i].cycles, df[df.unit==i].S7)
plt.ylabel('Sensor measurements')
plt.xlabel('# cycles')

I'd like to create a subplot to show all the sensors. I have attempted by using iteration but it doesnt work.

sensors = ["Op1", "Op2", "Op3", "S2", "S3", "S4", "S5", "S6", "S7", "S8", "S9", "S10", "S11",
           "S12", "S13", "S14", "S15", "S16", "S17", "S18", "S19", "S20", "S21"]
i = 1
for sensor in sensors:
    for n in range(1,101):
        plt.subplot(len(sensors), 1, i)
        plt.plot(df[df.unit==n].cycles, df[df.unit==n].sensor)
    i += 1

What are the changes I should apply to my code? Thank you so much

Quang Hoang · Accepted Answer · 2020-06-30 14:56:34Z

1

You can create a list of subplots first, and plot into them:

fig, axes = plt.subplots(2, 11)   # change these numbers as wished

for sensor, ax in zip(sensors, axes.ravel()):
    for n in range(1,101):
        df[df.unit==n].plot(x='cycles', y=sensor, ax=ax)
        ax.set_title(sensor)

        # remove the long legend
        ax.legend().remove()

edited Jun 30, 2020 at 14:56

answered Jun 30, 2020 at 14:14

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

4 Comments

Luqman Buang Over a year ago

i get a weird concoction of plots: imgur.com/a/Tz0oWVg did i input it wrong?

Quang Hoang Over a year ago

You got it correctly, it's just a small image. Try to pass figsize=(20,4) into plt.subplots.

Quang Hoang Over a year ago

@LuqmanBuang also see update for removal of the unnecessary long legend in each axis.

Luqman Buang Over a year ago

thank you! By the way there is a small error in the for loop; change the 'i' to 'n' and its all set! :D

Collectives™ on Stack Overflow

How do i iterate my dataframe values to create subplots?

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related