Comment tracer les valeurs d'un tableau (ou matrice) en fonction d'un axe donné avec numpy en python ?

Exemple de comment tracer les valeurs d'un tableau (ou matrice) en fonction d'un axe donné avec numpy en python

Créer une simple matrice A

Pour illustrer le méthode, commençons par créer une matrice A:

import numpy as np

A = np.array([[1, 2.2, 3.3, 3.6, 5.1],
              [1.2, 2.3, 3.4, 4.1, 5.5],
              [0.8, 1.7, 3.1, 4.2, 5.6],
              [1.1, 2.4, 3.5, 4.5, 5.2],
              [1, 2, 3, 4, 5]])

utiliser meshgrid() pour obtenir les indices des éléments de A

xx, yy = np.meshgrid(np.arange(A.shape[1]),np.arange(A.shape[0]))

print(xx)

donne

[[0 1 2 3 4]
 [0 1 2 3 4]
 [0 1 2 3 4]
 [0 1 2 3 4]
 [0 1 2 3 4]]

Tracer avec matplotlib:

plt.scatter(xx.ravel(),A.ravel())

plt.ylabel('Array Values')
plt.xlabel('Axis 1 Indices')

plt.savefig("array_values_vs_indices.png", bbox_inches='tight', dpi=100)

plt.show()

Liens

• What is the purpose of meshgrid in Python / NumPy?