Python >> Tutoriel Python >  >> Python Tag >> NumPy

Comment obtenir la forme d'un tableau numpy ?

NumPy est une bibliothèque Python populaire pour la science des données. La bibliothèque se concentre sur les calculs sur les tableaux, les vecteurs et les matrices. Si vous travaillez avec des données, il est impossible d'éviter NumPy.

Dans ce didacticiel, vous en apprendrez plus sur la forme d'un tableau NumPy. En particulier, vous allez apprendre la solution à la question suivante :

Comment obtenir la forme d'un tableau numpy ?

Répondre :Vous pouvez accéder à la forme d'un tableau NumPy via l'attribut tableau array.shape .

Pour obtenir la forme d'un tableau NumPy à n dimensions arr , appelez le arr.shape qui renvoie un tuple avec n valeurs, une par dimension. Chaque valeur de tuple donne le nombre d'éléments le long de cette dimension. Par exemple, un tableau à deux dimensions avec x lignes et y les colonnes ont la forme (x,y) .

Voici un exemple de tableau à deux dimensions :

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6]])

print(arr.shape) 
# Output: (2, 3)

Voici un exemple de tableau unidimensionnel :

import numpy as np

arr = np.array([1, 2, 3])

print(arr.shape) 
# Output: (3,)

Et voici un exemple de tableau tridimensionnel :

import numpy as np

arr = np.array([[[1, 2], [3, 4]],
                [[5, 6], [7, 8]]])

print(arr.shape) 
# Output: (2, 2, 2)

Maintenant, testons et entraînons vos compétences en matière de forme NumPy :

Puzzle de forme NumPy

Pouvez-vous résoudre le casse-tête suivant concernant l'attribut de forme de NumPy ?

import numpy as np

# salary in ($1000) [2015, 2016, 2017]
dataScientist = [133, 132, 137]
productManager = [127, 140, 145]
designer = [118, 118, 127]
softwareEngineer = [129, 131, 137]

a = np.array([dataScientist,
              productManager,
              designer,
              softwareEngineer])

print(a.shape[0])
print(a.shape[1])

Exercice :Quel est le résultat de ce puzzle ?

Vous pouvez résoudre ce puzzle de manière interactive sur notre application Finxter ici :

Nous travaillons sur les données salariales de quatre métiers :

  • scientifiques des données,
  • chefs de produit,
  • concepteurs, et
  • ingénieurs en logiciel.

Nous créons quatre listes qui stockent le salaire annuel moyen des quatre emplois en milliers de dollars pour les trois années suivantes.

Ensuite, nous fusionnons ces quatre listes dans un tableau à deux dimensions (noté matrice ), c'est-à-dire une liste de listes. Chaque liste de salaire d'un seul emploi devient une ligne de cette matrice. Chaque ligne comporte trois colonnes, une pour chaque année.

Le puzzle imprime la forme de cette matrice qui est le nombre d'éléments dans chaque dimension. Par exemple, une matrice avec n lignes et m les colonnes ont la forme (n,m) . Comme notre matrice bidimensionnelle a 4 lignes et 3 colonnes, la solution de ce puzzle est 4 et 3.

Voulez-vous devenir un maître NumPy ? Découvrez notre livre de puzzle interactif Coffee Break NumPy et boostez vos compétences en data science ! (Le lien Amazon s'ouvre dans un nouvel onglet.)