matplotlibで自由自在にsubplotする。

MATLABではsubplot(1.5,1,1)とかやるとfigureの2/3を占めるようなsubplotを作ることができて、便利だったんだけどmatplotlibでどうやるのかいつも気になっていたのですが、今日とうとうやり方を発見。Scipy lecture notesのsubplotの項 に、あった。

gridspecというのがあって、これでグリッドを作る。

import matplotlib.gridspec as gridspec
pl.figure(figsize=(6, 4))
G = gridspec.GridSpec(3, 3)

で、axisを指定するときに

axes_3 = pl.subplot(G[1:, -1])

のようにしてグリッドをnumpy arrayをスライスする感じで指定するらしい。非常にエレガントな方法。MATLABで小数点絡んだやり方は非常に混乱したのを覚えている。

横に長いプロットを２つに縦長を１つをpylabの例でやります。

from pylab import *

toy_data = np.random.rand(100)

G = gridspec.GridSpec(2,3)

subplot(G[0,:2])
plot(toy_data)

subplot(G[1,:2])
plot(toy_data ** 2, 'r')

subplot(G[:,2])
bar(0.6, np.mean(toy_data ** 2), facecolor='none')
xlim([0,2])

できました！バーグラフのY軸がめり込んでいるのはpylabならtight_layout()で一発でレイアウトが自動で最適化されます。

それにしてもこのScipy lecture notesは他にも非常に充実していて、あとできっちり読んだほうが良さそう。色々と評判がよいscikit-learnまでカバーしてあるのが嬉しい。