Comment déterminer le nombre de dimensions d'une matrice en python ?

Exemple de comment déterminer le nombre de dimensions d'une matrice en python:

Déterminer le nombre de dimensions d'une matrice

Soit la matrice suivante

>>> import numpy as np
>>> A = np.array([[3,9,4],[5,8,1],[9,2,5]])
>>> A
array([[3, 9, 4],
       [5, 8, 1],
       [9, 2, 5]])

Pour déterminer la forme d'une matrice on peut alors utiliser shape:

>>> A.shape
(3, 3)

et en déduire le nombre de dimensions avec la fonction len:

>>> len(A.shape)
2

Exemple de cas

Soit une matrice input_x de dimension 1 ou 2.

Transposer la matrice input_x, si la matrice input_x est de dimension 2 et que le nombre de colonnes > nombre de lignes:

>>> def data_preparation(input_x):
...     input_x_shape = input_x.shape
...     if len(input_x_shape) == 2:
...             if input_x_shape[1] > input_x_shape[0]: 
...                     input_x = input_x.T
...     return input_x
... 
>>> input_x = np.arange(9)
>>> input_x
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
>>> data_preparation(input_x)
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
>>> input_x = np.arange(20)
>>> input_x = input_x.reshape(2,10)
>>> data_preparation(input_x)
array([[ 0, 10],
       [ 1, 11],
       [ 2, 12],
       [ 3, 13],
       [ 4, 14],
       [ 5, 15],
       [ 6, 16],
       [ 7, 17],
       [ 8, 18],
       [ 9, 19]])
>>>

Références