Comment récupérer des tableaux NumPy et des listes Python à partir des index et des colonnes Pandas ?

Récupération de tableaux NumPy à partir d'index et de colonnes Pandas

Lorsque vous travaillez avec Pandas, accéder à l'index ou à des colonnes spécifiques sous forme de tableaux NumPy ou de listes Python est une exigence courante. Voici comment y parvenir :

Conversion en tableau NumPy

Pour obtenir un tableau NumPy, exploitez l'attribut valeurs. Cet attribut vous permet de récupérer les données sous-jacentes sans avoir besoin de conversion explicite :

<code class="python">import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['a', 'b', 'c'])

# Get the index as a NumPy array
index_array = df.index.values
print(index_array) 
# Output: array(['a', 'b', 'c'], dtype=object)</code>

Conversion en liste Python

Pour récupérer l'index ou les colonnes sous forme de listes Python, utilisez la tolist() method :

<code class="python"># Get the index as a list
index_list = df.index.tolist()
print(index_list) # Output: ['a', 'b', 'c']</code>

De même, vous pouvez obtenir les colonnes sous forme de listes en accédant à l'attribut columns puis en appliquant la méthode 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!