Comment convertir un DataFrame Pandas en un tableau NumPy ?

Convertir une trame de données pandas en tableau NumPy

Comment convertir une trame de données pandas en tableau NumPy ?

Utiliser df.to_numpy()

df.to_numpy() est la méthode recommandée pour convertir une trame de données pandas en un tableau NumPy. Il est défini sur les objets Index, Series et DataFrame et fournit un moyen cohérent d'extraire le tableau NumPy sous-jacent.

import pandas as pd
import numpy as np

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# Convert the entire DataFrame
array = df.to_numpy()

# Convert specific columns
array = df[['A', 'C']].to_numpy()

Le tableau renvoyé est une vue du DataFrame d'origine, donc toutes les modifications apportées au tableau sera reflété dans le DataFrame. Pour obtenir une copie à la place, utilisez to_numpy(copy=True).

Préserver les types

Si vous devez conserver les types dans le résultat, vous pouvez utiliser DataFrame .to_records() au lieu de to_numpy().

record_array = df.to_records()

Cependant, notez que to_records() renvoie un rec.array, qui n'est pas un tableau NumPy. Au lieu de cela, il s'agit d'un tableau structuré qui s'apparente davantage à un DataFrame pandas. Si vous avez besoin d'un vrai tableau NumPy avec des types préservés, vous pouvez utiliser np.rec.fromrecords() :

named_array = np.rec.fromrecords(v, names=v.columns.tolist())

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!