Comment extraire les en-têtes de colonnes de Pandas DataFrame avec la saisie de l'utilisateur ?

Extraire les en-têtes de colonnes d'un Pandas DataFrame

Obtenir une liste d'en-têtes de colonnes à partir d'un Pandas DataFrame est une opération courante pour l'analyse des données. Dans cet article, nous montrerons comment y parvenir lorsque le DataFrame est généré à partir de l'entrée de l'utilisateur, garantissant la compatibilité avec un nombre ou des noms de colonnes inconnus.

Extraction d'en-tête de colonne DataFrame

Pour acquérir la liste des en-têtes de colonnes d'un DataFrame, vous pouvez utiliser ce qui suit :

• columns.values : Cet attribut renvoie un tableau d'étiquettes de colonnes, qui peuvent être converti en liste à l'aide de list(my_dataframe.columns.values).
• Diffusion directe : Alternativement, vous pouvez simplement convertir le DataFrame en liste avec list(my_dataframe) Cela entraînera une liste d'en-têtes de colonnes suivis des valeurs DataFrame.

Exemple

Considérez le DataFrame suivant :

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

data = {
    'y': [1, 2, 8, 3, 6, 4, 8, 9, 6, 10],
    'gdp': [2, 3, 7, 4, 7, 8, 2, 9, 6, 10],
    'cap': [5, 9, 2, 7, 7, 3, 8, 10, 4, 7]
}

df = pd.DataFrame(data)</code>

Obtention des en-têtes de colonnes

À l'aide de la méthode columns.values :

<code class="python">headers = list(df.columns.values)
print(headers)

# Output: ['y', 'gdp', 'cap']</code>

Utilisation du casting direct :

<code class="python">headers = list(df)
print(headers)

# Output: ['y', 'gdp', 'cap']</code>

Les deux approches fourniront une liste d'en-têtes de colonnes : ['y', 'pib', 'cap'].

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!