Pandas DataFrame Plot: lists of values

Question

I have a Pandas DataFrame which resembles:

pd.DataFrame({
    'el1': {
        'steps': [0,1,2], 
        'values': [10, 9, 8]
    },
    'el2': {
        'steps': [0,1,2], 
        'values': [1,  2, 8]
    },
    'el3': {
        'steps': [0,1,2], 
        'values': [5,  9, 4]
    }
})


        el1         el2         el3
steps   [0, 1, 2]   [0, 1, 2]   [0, 1, 2]
values  [10, 9, 8]  [1, 2, 8]   [5, 9, 4]

what would be the best way to try and use Panda's DataFrame plot to get some simple line plots with values on the y axis and steps on the x axis? (e.g. there should be three lines)

alec_djinn · Accepted Answer · 2019-03-06 09:57:00Z

1

Use matplotlib

c = ['r', 'g', 'b']
for i in range(df.shape[1]):
    plt.plot(df.iloc[0, i], df.iloc[1, i], c=c[i], label=df.columns[i])
plt.legend(df.columns)
plt.xlabel('Steps')
plt.ylabel('Values')
plt.show()

edited Mar 6, 2019 at 9:57

alec_djinn

10.9k9 gold badges57 silver badges77 bronze badges

answered Mar 6, 2019 at 9:20

meW

3,97710 silver badges27 bronze badges

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

2 Comments

kerwei Over a year ago

Sorry, I meant to edit my own answer but edited yours instead. Not sure how to reject it xD

meW Over a year ago

@kerwei I've taken care of it.

kerwei · Accepted Answer · 2019-03-06 09:55:34Z

Here's a different construction. Largely because I'm more comfortable with transforming the underlying data frame first before plotting. The plotting of the graph is pretty much the same so, credits go to @meW for the lines of code.

import pandas as pd
import matplotlib.pyplot as plt


df = pd.DataFrame({
    'el1': {
        'steps': [0,1,2], 
        'values': [10, 9, 8]
    },
    'el2': {
        'steps': [0,1,2], 
        'values': [1,  2, 8]
    },
    'el3': {
        'steps': [0,1,2], 
        'values': [5,  9, 4]
    }
})

ndf = pd.DataFrame({v:df[v].values[1] for v in df.columns})
ndf.index = df.el1.steps
ndf.columns = df.columns

>>>ndf
   el1  el2  el3
0   10    1    5
1    9    2    9
2    8    8    4

plt.plot(ndf)
plt.legend(ndf.columns)
plt.xlabel('Steps')
plt.ylabel('Values')
plt.show()

Collectives™ on Stack Overflow

Pandas DataFrame Plot: lists of values

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related