Jak uzyskać nazwy kolumn w pandach (3 metody)

Możesz użyć następujących metod, aby uzyskać nazwy kolumn w ramce DataFrame pand:

Metoda 1: Uzyskaj wszystkie nazwy kolumn

 list (df)

Metoda 2: Pobierz nazwy kolumn alfabetycznie

 sorted (df)

Metoda 3: Uzyskaj nazwy kolumn z określonym typem danych

 list ( df.select_dtypes (include=[' int64 ', ' bool ']))

Poniższe przykłady pokazują, jak używać każdej metody z następującą ramką DataFrame pand:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F'],
                   ' points ': [18, 22, 19, 14, 14, 11],
                   ' assists ': [5, 7, 7, 9, 12, 9],
                   ' playoffs ': [True, False, False, True, True, True]})

#view DataFrame
print (df)

  team points assists playoffs
0 A 18 5 True
1 B 22 7 False
2 C 19 7 False
3 D 14 9 True
4 E 14 12 True
5 F 11 9 True

Przykład 1: Pobierz wszystkie nazwy kolumn

Najłatwiejszym sposobem uzyskania wszystkich nazw kolumn w ramce DataFrame pand jest użycie list() w następujący sposób:

 #get all column names
list (df)

['team', 'points', 'assists', 'playoffs']

Wynikiem jest lista zawierająca nazwy czterech kolumn ramki DataFrame pand.

Przykład 2: Pobierz nazwy kolumn alfabetycznie

Aby uzyskać nazwy kolumn w ramce danych pand w kolejności alfabetycznej, możesz użyć funkcji sorted() w następujący sposób:

 #get column names in alphabetical order
sorted (df)

['assists', 'playoffs', 'points', 'team']

Wynikiem jest lista zawierająca nazwy czterech kolumn ramki DataFrame pand, wymienione w kolejności alfabetycznej.

Możesz także użyć argumentu Reverse=True , aby uzyskać nazwy kolumn w odwrotnej kolejności alfabetycznej:

 #get column names in reverse alphabetical order
sorted (df, reverse= True )

['team', 'points', 'playoffs', 'assists']

Przykład 3: Uzyskaj nazwy kolumn z określonym typem danych

Aby wyświetlić typ danych każdej kolumny w ramce DataFrame, możesz użyć następującej składni:

 #view data type of each column
df. dtypes

team object
int64 dots
assists int64
playoffs bool
dtype:object

Następnie możesz użyć funkcjiselect_dtypes() , aby uzyskać tylko nazwy kolumn z określonym typem danych.

Na przykład możemy użyć następującej składni, aby uzyskać tylko nazwy kolumn, które mają typ danych int64 lub bool :

 #get all columns that have data type of int64 or bool
list ( df.select_dtypes (include=[' int64 ', ' bool ']))

['points', 'assists', 'playoffs']

Wynikiem jest lista nazw kolumn mających typ danych int64 lub bool .

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje na pandach:

Pandy: jak zmienić nazwy kolumn
Pandy: Jak ustawić kolumnę jako indeks
Pandy: pobierz indeks wierszy, których kolumna odpowiada wartości