Tech - calcul matriciel avec numpy#

Links: notebook, html, python, slides, GitHub

numpy est la librairie incontournable pour faire des calculs en Python. Ces fonctionnalités sont disponibles dans tous les langages et utilisent les optimisations processeurs. Il est hautement improbable d’écrire un code aussi rapide sans l’utiliser.

numpy implémente ce qu’on appelle les opérations matricielles basiques ou plus communément appelées BLAS. Quelque soit le langage, l’implémentation est réalisée en langage bas niveau (C, fortran, assembleur) et a été peaufinée depuis 50 ans au gré des améliorations matérielles.

from jyquickhelper import add_notebook_menu
add_notebook_menu()
%matplotlib inline

Enoncé#

La librairie numpy propose principalement deux types : array et matrix. Pour faire simple, prenez toujours le premier. Ca évite les erreurs. Les array sont des tableaux à plusieurs dimensions.

La maîtrise du slice#

Le slice est l’opérateur : (décrit sur la page indexing). Il permet de récupérer une ligne, une colonne, un intervalle de valeurs.

import numpy

mat = numpy.array([[0, 5, 6, -3],
                   [6, 7, -4, 8],
                   [-5, 8, -4, 9]])
mat
array([[ 0,  5,  6, -3],
       [ 6,  7, -4,  8],
       [-5,  8, -4,  9]])
mat[:2], mat[:, :2], mat[0, 3], mat[0:2, 0:2]
(array([[ 0,  5,  6, -3],
        [ 6,  7, -4,  8]]),
 array([[ 0,  5],
        [ 6,  7],
        [-5,  8]]),
 -3,
 array([[0, 5],
        [6, 7]]))

La maîtrise du nan#

nan est une convention pour désigner une valeur manquante. Elle réagit de façon un peut particulière. Elle n’est égale à aucune autre y compris elle-même.

numpy.nan == numpy.nan
False
numpy.nan == 4
False

Il faut donc utiliser une fonction spéciale isnan.

numpy.isnan(numpy.nan)
True

La maîtrise des types#

Un tableau est défini par ses dimensions et le type unique des éléments qu’il contient.

matint = numpy.array([0, 1, 2])
matint.shape, matint.dtype
((3,), dtype('int32'))

C’est le même type pour toute la matrice. Il existe plusieurs type d’entiers et des réels pour des questions de performances.

%timeit matint * matint
508 ns ± 44.7 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
matintf = matint.astype(numpy.float64)
matintf.shape, matintf.dtype
((3,), dtype('float64'))
%timeit matintf * matintf
484 ns ± 30.4 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
%timeit matintf * matint
894 ns ± 78.8 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

Un changement de type et le calcul est plus long.

La maîtrise du broadcasting#

Le broadcasting signifie que certaines opérations ont un sens même si les dimensions des tableaux ne sont pas tout à fait égales.

mat
array([[ 0,  5,  6, -3],
       [ 6,  7, -4,  8],
       [-5,  8, -4,  9]])
mat + 1000
array([[1000, 1005, 1006,  997],
       [1006, 1007,  996, 1008],
       [ 995, 1008,  996, 1009]])
mat + numpy.array([0, 10, 100, 1000])
array([[   0,   15,  106,  997],
       [   6,   17,   96, 1008],
       [  -5,   18,   96, 1009]])
mat + numpy.array([[0, 10, 100]]).T
array([[  0,   5,   6,  -3],
       [ 16,  17,   6,  18],
       [ 95, 108,  96, 109]])

La maîtrise des index#

mat = numpy.array([[0, 5, 6, -3],
                   [6, 7, -4, 8],
                   [-5, 8, -4, 9]])
mat
array([[ 0,  5,  6, -3],
       [ 6,  7, -4,  8],
       [-5,  8, -4,  9]])
mat == 5
array([[False,  True, False, False],
       [False, False, False, False],
       [False, False, False, False]])
mat == numpy.array([[0, -4, 9]]).T
array([[ True, False, False, False],
       [False, False,  True, False],
       [False, False, False,  True]])
(mat == numpy.array([[0, -4, 9]]).T).astype(numpy.int64)
array([[1, 0, 0, 0],
       [0, 0, 1, 0],
       [0, 0, 0, 1]], dtype=int64)
mat * (mat == numpy.array([[0, -4, 9]]).T).astype(numpy.int64)
array([[ 0,  0,  0,  0],
       [ 0,  0, -4,  0],
       [ 0,  0,  0,  9]], dtype=int64)

La maîtrise des fonctions#

On peut regrouper les opérations que numpy propose en différents thèmes. Mais avant il

Q1 : calculer la valeur du \chi_2 d’un tableau de contingence#

La formule est . Et il faut le faire sans boucle. Vous pouvez comparer avec la fonction chisquare de la librairie scipy qui est une extension de numpy.

\chi_2 = N \sum_{i,j} p_{i.} p_{.j} \left( \frac{\frac{O_{ij}}{N} - p_{i.} p_{.j}}{p_{i.} p_{.j}}\right)^2

Q2 : calculer une distribution un peu particulière#

La fonction histogram permet de calculer la distribution empirique de variables. Pour cette question, on tire un vecteur aléatoire de taille 10 avec la fonction rand, on les trie par ordre croissant, on recommence plein de fois, on calcule la distribution du plus grand nombre, du second plus grand nombre, …, du plus petit nombre.

Q3 : on veut créer une matrice identité un million par un million#

Vous pouvez essayer sans réfléchir ou lire cette page d’abord : csr_matrix.

Q4 : vous devez créer l’application StopCovid#

Il existe une machine qui reçoit la position de 3 millions de téléphones portable. On veut identifier les cas contacts (rapidement).

Réponses#

Q1 : calculer la valeur du \chi_2 d’un tableau de contingence#

La formule est . Et il faut le faire sans boucle. Vous pouvez comparer avec la fonction chisquare de la librairie scipy qui est une extension de numpy.

\chi_2 = N \sum_{i,j} p_{i.} p_{.j} \left( \frac{\frac{O_{ij}}{N} - p_{i.} p_{.j}}{p_{i.} p_{.j}}\right)^2

import numpy
O = numpy.array([[15., 20., 13.], [4., 9., 5.]])
O
array([[15., 20., 13.],
       [ 4.,  9.,  5.]])
def chi_square(O):
    N = numpy.sum(O)
    pis = numpy.sum(O, axis=1, keepdims=True) / N
    pjs = numpy.sum(O, axis=0, keepdims=True) / N
    pispjs = pis @ pjs
    chi = pispjs * ((O / N - pispjs) / pispjs) ** 2
    return numpy.sum(chi) * N

chi_square(O)
0.5798254016266716

Q2 : calculer une distribution un peu particulière#

La fonction histogram permet de calculer la distribution empirique de variables. Pour cette question, on tire un vecteur aléatoire de taille 10 avec la fonction rand, on les trie par ordre croissant, on recommence plein de fois, on calcule la distribution du plus grand nombre, du second plus grand nombre, …, du plus petit nombre.

rnd = numpy.random.rand(10)
rnd
array([0.98556467, 0.47377301, 0.77148185, 0.26135908, 0.27373018,
       0.0240458 , 0.55360714, 0.3575369 , 0.71740732, 0.3260206 ])
numpy.sort(rnd)
array([0.0240458 , 0.26135908, 0.27373018, 0.3260206 , 0.3575369 ,
       0.47377301, 0.55360714, 0.71740732, 0.77148185, 0.98556467])
def tirage(n):
    rnd = numpy.random.rand(n)
    trie = numpy.sort(rnd)
    return trie[-1]

tirage(10)
0.876020129318981
def plusieurs_tirages(N, n):
    rnd = numpy.random.rand(N, n)
    return numpy.max(rnd, axis=1)

plusieurs_tirages(5, 10)
array([0.99594032, 0.67914189, 0.98105965, 0.93181536, 0.86827764])
t = plusieurs_tirages(5000, 10)
hist = numpy.histogram(t)
hist
(array([   5,    8,   20,   35,  111,  221,  407,  785, 1273, 2135],
       dtype=int64),
 array([0.437878  , 0.49408914, 0.55030028, 0.60651142, 0.66272256,
        0.7189337 , 0.77514485, 0.83135599, 0.88756713, 0.94377827,
        0.99998941]))
import matplotlib.pyplot as plt
plt.plot(hist[1][1:], hist[0] / hist[0].sum());
../_images/2020_numpy_50_0.png

Q3 : on veut créer une matrice identité un million par un million#

Vous pouvez essayer sans réfléchir ou lire cette page d’abord : csr_matrix.

(10^6)^2=10^{12}>10 Go, bref ça ne tient pas en mémoire sauf si on a une grosse machine. Les matrices creuses (ou sparses en anglais), sont adéquates pour représenter des matrices dont la grande majorité des coefficients sont nuls car ceux-ci ne sont pas stockés. Concrètement, la matrice enregistre uniquement les coordonnées des coefficients et les valeurs non nuls.

import numpy
from scipy.sparse import csr_matrix
ide = csr_matrix((1000000, 1000000), dtype=numpy.float64)
ide.setdiag(1.)
c:python395_x64libsite-packagesscipysparse_index.py:125: SparseEfficiencyWarning: Changing the sparsity structure of a csr_matrix is expensive. lil_matrix is more efficient.
  self._set_arrayXarray(i, j, x)

Q4 : vous devez créer l’application StopCovid#

Il existe une machine qui reçoit la position de 3 millions de téléphones portable. On veut identifier les cas contacts (rapidement).

Si on devait calculer toutes les paires de distance, cela prendrait un temps fou. Il faut ruser. Le plus simple est de construire une grille sur le territoire français puis d’associer à chaque téléphone portable la grille dans laquelle il se trouve. Dans une cellule de la grille, le nombre de paires est beaucoup plus réduit. Ce n’est pas la seule astuce qu’il faudra utiliser. Mais c’est un bon début.