Showing different data using dataframe.plot from pandas

Question

I want to show some data using plots. My dataframe has 3 columns,

Id  Round Time
1   1     158
1   2     169
1   3     172
2   1     156
2   2     168
2   3     176

With this example what I want to do is show 2 lines in my plot (with different colors), each line must correspond to the id, the x axis with the number of the round and the y axis with time.

Now I'm doing this and it's obviously wrong,

import pandas as pd

data = [[1,1,158],[1,2,169],[1,3,172],
        [2,1,156],[2,2,168],[2,3,176]]
cols = ['Id','Round','Time']
df = pd.DataFrame(data, columns=cols)

df.plot(kind='line',x='Round',y='Time')
plt.show()

How can I differentiate each of the different ids?

Thank you very much!!

Rajat Mishra · Accepted Answer · 2020-04-15 12:39:43Z

2

You need to create a pivot and then plot it.

In [49]: df.pivot('Round','Id','Time').plot(kind='line')
Out[49]: <matplotlib.axes._subplots.AxesSubplot at 0x2362f31ae10>

In [50]: plt.show()

answered Apr 15, 2020 at 12:39

Rajat Mishra

3,7904 gold badges32 silver badges44 bronze badges

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

Collectives™ on Stack Overflow

Showing different data using dataframe.plot from 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