Python: How to draw some boxplots for data frames using matplotlib?

Question

I have a Pandas dataframe with 4 columns - id, label, var1 and var2.

From this data, I would like to draw two box plots each for var1 and var2 - one for where the label==0 and the other for where label==1.

Sample dataframe:

+-----+-----+----+----+
| id  |label|var1|var2|
+-----+-----+----+----+
| 1   |  0  | 0.7| 0.6|
| 2   |  0  | 0.3| 0.4|
| 3   |  1  | 0.2| 0.2|
| 4   |  1  | 0.8| 0.1|
| 5   |  0  | 0.0| 0.9|
+-----+-----+----+----+

Code to generate dataframe:

l=[(1,0,0.7,0.6),(2,0,0.3,0.4),(3,1,0.2,0.2),(4,1,0.8,0.1),(5,0,0.0,0.9)]
names=["id","label", "var1", "var2"]
db=sqlContext.createDataFrame(l,names)
db.show()
db = db.toPandas()

How can I implement this using matplotlib?

ImportanceOfBeingErnest · Accepted Answer · 2017-10-24 18:53:56Z

4

Just use df.boxplot() on your dataframe.

import pandas as pd
import matplotlib.pyplot as plt

l=[(1,0,0.7,0.6),(2,0,0.3,0.4),(3,1,0.2,0.2),(4,1,0.8,0.1),(5,0,0.0,0.9)]
names=["id","label", "var1", "var2"]
df = pd.DataFrame(l, columns=names)

df.boxplot(column=["var1", "var2"],by="label")

plt.show()

answered Oct 24, 2017 at 18:53

ImportanceOfBeingErnest

342k61 gold badges737 silver badges771 bronze badges

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

2 Comments

Thaise Over a year ago

Thanks! Do you know how can I set y-axis limit in matplotlib?

ImportanceOfBeingErnest Over a year ago

Yes, set y-axis limit in matplotlib.

Collectives™ on Stack Overflow

Python: How to draw some boxplots for data frames using matplotlib?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related