plotting dataframe values in loop

Question

I have a csv file named as data.csv and it contain [10000 rows x 1 columns] and i read the data as

a=read_csv("data.csv")

print(a)

                                                  dt
0     3.5257,3.5257,3.5257,3.5257,3.5257,3.5257,3.52...
1     3.3414,3.3414,3.3414,3.3414,3.3414,3.3414,3.34...
2     3.5722,3.5722,3.5722,3.5722,3.5722,3.5722,3.57...
.....

Now I want to plot each indexed rows in loop and want to save each plot in the name of index values . For example

i need to produce 0.jpg by using o indexed values ---->  3.5257,3.5257,3.5257,3.5257,3.5257,3.5257,3.52...
similarly 1.jpg by using 1 index values---->  3.3414,3.3414,3.3414,3.3414,3.3414,3.3414,3.34...

My code

a=read_csv("data.csv")
print(a)
for data in a:
    plt.plot(data)

Hope experts will help me providing the answer.Thanks.

Corralien · Accepted Answer · 2023-01-28 09:04:17Z

1

Use savefig to create image file:

import matplotlib.pyplot as plt

for name, data in a['dt'].str.split(',').items():
    plt.plot(data)
    plt.savefig(f'{name}.jpg')
    plt.close()

0.jpg

1.jpg

answered Jan 28, 2023 at 9:04

Corralien

121k8 gold badges43 silver badges68 bronze badges

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

Comments

Laurent · Accepted Answer · 2023-01-28 09:02:18Z

1

With the following file.csv example:

3.5257,3.5257,3.5257,3.5257,3.5257,3.5257,3.52
3.3414,3.3414,3.3414,3.3414,3.3414,3.3414,3.34
3.5722,3.5722,3.5722,3.5722,3.5722,3.5722,3.57

Here is one way to do it with Pandas iterrows and Python f-strings:

import matplotlib.pyplot as plt
import pandas as pd


df = pd.read_csv("../file.csv", header=None)

for i, row in df.iterrows():
    fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(8, 4))
    ax.plot(row)
    fig.savefig(f"./{i}.png")

answered Jan 28, 2023 at 9:02

Laurent

13.7k7 gold badges30 silver badges49 bronze badges

Collectives™ on Stack Overflow

plotting dataframe values in loop

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related