Як отримати назви стовпців у pandas (3 методи)

Ви можете використовувати такі методи, щоб отримати назви стовпців у pandas DataFrame:

Спосіб 1: отримати всі назви стовпців

 list (df)

Спосіб 2. Отримайте назви стовпців в алфавітному порядку

 sorted (df)

Спосіб 3. Отримайте назви стовпців із певним типом даних

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

У наведених нижче прикладах показано, як використовувати кожен метод із такими pandas DataFrame:

 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

Приклад 1: отримати всі назви стовпців

Найпростіший спосіб отримати всі назви стовпців у pandas DataFrame — це використати list() наступним чином:

 #get all column names
list (df)

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

Результатом буде список, що містить чотири назви стовпців pandas DataFrame.

Приклад 2. Отримайте назви стовпців в алфавітному порядку

Щоб отримати назви стовпців у pandas DataFrame в алфавітному порядку, ви можете використати функцію sorted() таким чином:

 #get column names in alphabetical order
sorted (df)

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

Результатом буде список, що містить чотири назви стовпців pandas DataFrame, перераховані в алфавітному порядку.

Ви також можете використовувати аргумент reverse=True , щоб отримати назви стовпців у зворотному алфавітному порядку:

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

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

Приклад 3. Отримання імен стовпців із певним типом даних

Ви можете використовувати такий синтаксис, щоб відобразити тип даних кожного стовпця в DataFrame:

 #view data type of each column
df. dtypes

team object
int64 dots
assists int64
playoffs bool
dtype:object

Потім ви можете використовувати функцію select_dtypes() , щоб отримати лише назви стовпців із певним типом даних.

Наприклад, ми можемо використовувати такий синтаксис, щоб отримати лише імена стовпців, які мають тип даних int64 або bool :

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

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

Результатом є список імен стовпців із типом даних int64 або bool .

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:

Pandas: як перейменувати стовпці
Pandas: Як встановити стовпець як індекс
Pandas: отримати індекс рядків, стовпець яких відповідає значенню