#coding: utf-8

import pylab as pl, numpy as np
import cPickle, gzip

mnist = cPickle.load(gzip.open("mnist.pkl.gz", "rb"))

start = 10
end = 15

for index in range(start, end + 1):

    # mnistデータから1枚持ってくる
    a_digit = mnist[0][0][index]
    # 答えを出力
    print mnist[0][1][index]

    # 1次元なので，2次元のリストに変換
    arr = []
    for i in range(28):
        arr.append(a_digit[(i*28):(i*28)+28])

    # オフスクリーンにプロット
    pl.imshow(arr, interpolation='nearest', cmap='bone', origin='lower')
    #pl.colorbar(shrink=.92)
    pl.colorbar()

    # 軸の反転
    ax = pl.gca()
    #ax.invert_xaxis()
    ax.invert_yaxis()

    # 表示
    pl.show()