Comment convertir une série ou un index Pandas en tableau NumPy ?

Question

Comment obtenir l'index ou la colonne d'un DataFrame sous la forme d'un tableau NumPy ou d'une liste Python ?

Answer 1

Une méthode plus récente consiste à utiliser la fonction .to_numpy().

Si j'ai un cadre de données avec une colonne "prix", je peux le convertir comme suit :

priceArray = df['price'].to_numpy()

Vous pouvez également transmettre le type de données, tel que float ou object, en tant qu'argument de la fonction

Answer 2

J'ai converti les pandas dataframe a list et a ensuite utilisé l'outil de base list.index() . Voici ce qu'il en est :

dd = list(zone[0]) #Where zone[0] is some specific column of the table
idx = dd.index(filename[i])

Vous avez votre valeur d'index comme idx .

Answer 3

Vous trouverez ci-dessous une méthode simple pour convertir une colonne de dataframe en un tableau NumPy.

df = pd.DataFrame(somedict)
ytrain = df['label']
ytrain_numpy = np.array([x for x in ytrain['label']])

ytrain_numpy est un tableau NumPy.

J'ai essayé avec to.numpy() mais j'ai obtenu l'erreur suivante :

TypeError : no supported conversion for types : (dtype('O'),)* while doing Binary Relevance classfication using Linear SVC.

to.numpy() convertissait le dataFrame en tableau NumPy, mais le type de données de l'élément interne était une liste, ce qui a provoqué l'erreur ci-dessus.