Exemple de comment déterminer le nombre de dimensions d'une matrice en python:
Table des matières
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
Liens | Site |
---|---|
shape | scipy doc |
size | scipy doc |
what does numpy ndarray shape do? | stackoverflow |
Python len() | programiz.com |