利用sklearn获取手写数字数据集，并进行可视化

Posted 2021-01-11

%matplotlib inline
from sklearn import datasets
from matplotlib import pyplot as plt

#获取数据集
digits = datasets.load_digits()

#可视化
for i in range(1, 11):
        plt.subplot(2, 5, i)  #划分成2行5列
        plt.imshow(digits.data[i - 1].reshape([8, 8]), cmap=plt.cm.gray_r)
        plt.text(3, 10, str(digits.target[i - 1])) #在图片的任意位置添加文本
        plt.xticks([]) #认为设置坐标轴显示的刻度值
        plt.yticks([])
plt.show()