Comment transformer une colonne Pandas en tableau et la transposer ?

Question

J'ai un cadre de données Pandas appelé "training_set" qui ressemble à la capture d'écran ci-dessous :

J'essaie de transformer la colonne "étiquette" en tableau et de la transposer. J'ai essayé de faire Y_train=np.asarray(training_set['label']) mais j'ai obtenu un tableau horizontal qui ressemble à la capture d'écran ci-dessous, ce qui n'est pas ce que je veux.

Je souhaite que le tableau s'affiche verticalement comme dans la capture d'écran ci-dessous (la capture d'écran comporte 2 variables par ligne). Le résultat que je souhaite obtenir ne doit contenir qu'une seule variable, le "label", par ligne).

Toute suggestion ou aide serait grandement appréciée !

Answer 1

Mise à jour v0.24

Utilisation DataFrame.to_numpy() c'est la nouvelle bonne façon d'extraire un tableau numpy :

training_set[['label']].to_numpy()

---

Ancienne réponse

Découpez votre colonne en une seule colonne DataFrame (en utilisant [[...]] ), et non en tant que Series :

Y_train = np.asarray(training_set[['label']])

Ou bien,

Y_train = training_set[['label']].values

Answer 2

Une autre solution consisterait à remodeler votre tableau pour qu'il prenne la forme suivante (-1,1) ce qui signifie "déduire le nombre de lignes, forcer à 1 colonne" :

Y_train = np.array(training_set['label']).reshape(-1,1)

Answer 3

D'une seule manière :

Y_train = training_set['label'].values[:, None]