Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Pandas: plotting two histograms on the same plot

I would like to have 2 histograms to appear on the same plot (with different colors, and possibly differente alphas). I tried

import random
x = pd.DataFrame([random.gauss(3,1) for _ in range(400)])
y = pd.DataFrame([random.gauss(4,2) for _ in range(400)])


x.hist( alpha=0.5, label='x')
y.hist(alpha=0.5, label='y')
x.plot(kind='kde', style='k--')
y.plot(kind='kde', style='k--')

plt.legend(loc='upper right')
plt.show()

This produces the result in 4 different plots. How can I have them on the same one?

like image 567
meto Avatar asked Nov 13 '14 15:11

meto


2 Answers

If I understood correctly, both hists should go into the same subplot. So it should be

fig = plt.figure()
ax = fig.add_subplot(111)
_ = ax.hist(x.values)
_ = ax.hist(y.values, color='red', alpha=.3)

You can also pass the pandas plot method an axis object, so if you want both kde's in another plot do:

fig = plt.figure()
ax = fig.add_subplot(111)
x.plot(kind='kde', ax=ax)
y.plot(kind='kde', ax=ax, color='red')

To get everything into a single plot you need two different y-scales since kde is density and histogram is frequency. For that you use the axes.twinx() command.

fig = plt.figure()
ax = fig.add_subplot(111)
_ = ax.hist(x.values)
_ = ax.hist(y.values, color='red', alpha=.3)

ax1 = ax.twinx()
x.plot(kind='kde', ax=ax1)
y.plot(kind='kde', ax=ax1, color='red')
like image 191
rustil Avatar answered Sep 23 '22 02:09

rustil


You can use plt.figure() and the function add_subplot(): the first 2 arguments are the number of rows and cols you want in your plot, the last is the position of the subplot in the plot.

fig = plt.figure()
subplot = fig.add_subplot(1, 2, 1)
subplot.hist(x.ix[:,0], alpha=0.5)
subplot = fig.add_subplot(1, 2, 2)
subplot.hist(y.ix[:,0], alpha=0.5)
like image 35
polku Avatar answered Sep 23 '22 02:09

polku