Bonjour,

Voici mon problème : j'importe des données depuis un fichier texte. Ce fichier texte peut contenir de 1 à plusieurs colonnes.

Avec l'aide de la fonction np.loadtxt je les importe directement dans une variable tel que :

data = np.loadtxt(...) où data est un array.

Donc quand je fais data.shape, ça me retourne le nombre de ligne et de colonne. par exemple :

>>> b=np.array([[1,2,3],[4,5,6]])
>>> b.shape
(2L, 3L)
>>> nb_col=int(b.shape[1])
3

Mais, si le array n'a qu'une seule colonne, ça marche plus :

>>> a=np.array([1,2,3])
>>> a.shape
(3L,)
>>> nb_col=int(a.shape[1])
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: tuple index out of range

Du coup comment faire pour savoir si j'ai un array qui a une colonne ou plus ??

Avec b.ndim, tu verrais combien de dimensions comportent ton array. Donc s'il n'y a qu'une seule colonne, il n'y aura qu'une dimension. Si tu as 2+ colonnes, tu auras plusieurs colonnes, et shape pourra te donner leur nombre.

Merci pour ta réponse. un peu sur le même principe, j'avais contourné le problème un peu comme tu le propose en faisant un len(b.shape).

J’espérais qu'une méthode à peine plus "propre" existait.

Merciiiii !!