Получение имени столбцов в Pandas DataFrame

Чтобы получить имена столбцов DataFrame, используйте свойство DataFrame.columns.

Синтаксис для использования свойства columns в DataFrame:

DataFrame.columns

Свойство columns возвращает объект типа Index. Мы могли получить доступ к отдельным именам, используя любую технику цикла в Python.

Пример 1: как распечатать?

В этом примере мы получим имена столбцов DataFrame и распечатаем их.

import pandas as pd

#initialize a dataframe
df = pd.DataFrame(
	[['Amol', 72, 67, 91],
	['Lini', 78, 69, 87],
	['Kiku', 74, 56, 88],
	['Ajit', 54, 76, 78]],
	columns=['name', 'physics', 'chemistry', 'algebra'])	

#get the dataframe columns
cols = df.columns 

#print the columns
print(cols)

Вывод:

Index(['name', 'physics', 'chemistry', 'algebra'], dtype='object')

Пример 2: с помощью индекса

Вы можете получить доступ к отдельным именам столбцов с помощью индекса.

import pandas as pd

#initialize a dataframe
df = pd.DataFrame(
	[['Amol', 72, 67, 91],
	['Lini', 78, 69, 87],
	['Kiku', 74, 56, 88],
	['Ajit', 54, 76, 78]],
	columns=['name', 'physics', 'chemistry', 'algebra'])	

#get the dataframe columns
cols = df.columns 

#print the columns
for i in range(len(cols)):
	print(cols[i])

Вывод:

name
physics
chemistry
algebra

Пример 3: с использованием цикла For Loop

Вы можете использовать цикл for для перебора столбцов DataFrame.

import pandas as pd

#initialize a dataframe
df = pd.DataFrame(
	[['Amol', 72, 67, 91],
	['Lini', 78, 69, 87],
	['Kiku', 74, 56, 88],
	['Ajit', 54, 76, 78]],
	columns=['name', 'physics', 'chemistry', 'algebra'])	

#get the dataframe columns
cols = df.columns 

#print the columns
for column in cols:
	print(column)

Вывод:

name
physics
chemistry
algebra

В этом руководстве по Pandas мы извлекли имена столбцов из DataFrame с помощью свойства DataFrame.column.