# -*- coding: utf-8 -*-
"""
Created on Mon Mar 13 20:30:54 2023

Données d'apprentissage pour kNN : second exercice
"""

import matplotlib.pyplot as plt
import numpy as np
from sklearn.datasets import load_digits
from numpy.random import randint


def obtenir_chiffres():
    digits = load_digits()
    '''Jeu de donnees Digits.'''
    X = digits.data
    Y = digits.target
    return X, Y


def montrer_chiffres(X):
    '''Montre des chiffres.'''
    plt.figure(randint(100000))
    nb = len(X)
    img = np.zeros((nb * 10, 10))
    for i in range(nb):
        img[i * 10 + 1:i * 10 + 9, 1:9] = X[i].reshape((8, 8))
    plt.imshow(img, cmap='binary')
    plt.xticks([])
    plt.yticks(5 + 10 * np.arange(nb), np.arange(nb))
    plt.show()