Tensorflow mnist basic

import numpy as np
import tensorflow as tf

import matplotlib.pyplot as plt
from tensorflow.examples.tutorials.mnist import input_data
from tensorflow.contrib.learn.python.learn.datasets import mnist
from matplotlib.cm import get_cmap

mnist = input_data.read_data_sets('data/', one_hot=True)
print type(mnist)
print mnist.train.num_examples
print mnist.test.num_examples

training = mnist.train.images
trainlabel = mnist.train.labels
testing = mnist.test.images
testlabel = mnist.test.labels

print type(training)
print training.shape
print type(testing)
print testing.shape
nsample = 1
randidx =  np.random.randint(training.shape[0], size=nsample)
# randidx =  np.random.choice(training.shape[0], nsample)
for i in randidx:
    curr_img = np.reshape(training[i,:], (28,28))
    curr_label = np.argmax(trainlabel[i,:])
#     plt.imshow(curr_img, cmap='gray')
    plt.matshow(curr_img, cmap=get_cmap('gray'))
    plt.title(str(i) + "th image wiht label:" + str(curr_label))
plt.show()