Как получить первый столбец Pandas DataFrame(с примерами)

Вы можете использовать следующий синтаксис, чтобы получить первый столбец кадра данных pandas:

df.iloc [:, 0]

Результатом является серия pandas. Если вы хотите, чтобы результатом был DataFrame pandas, вы можете использовать следующий синтаксис:

df.iloc [:, :1]

В следующих примерах показано, как использовать этот синтаксис на практике.

Пример 1: получить первый столбец Pandas DataFrame (возврат серии)

В следующем коде показано, как получить первый столбец кадра данных pandas:

import pandas as pd

#create DataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19, 23, 25, 29],
 'assists': [5, 7, 7, 9, 12, 9, 9, 4],
 'rebounds': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print(df)

 points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6
5 23 9 5
6 25 9 9
7 29 4 12

#get first column
first_col = df.iloc [:, 0]

#view first column
print(first_col)

0 25
1 12
2 15
3 14
4 19
5 23
6 25
7 29
Name: points, dtype: int64

Результатом является серия панд:

#check type of *first_col*
print( type (first_col))

<class 'pandas.core.series.Series'>

Пример 2: Получить первый столбец Pandas DataFrame(возврат DataFrame)

В следующем коде показано, как получить первый столбец кадра данных pandas и в результате вернуть кадр данных:

#get first column (and return a DataFrame)
first_col = df.iloc [:, :1]

#view first column
print(first_col)

 points
0 25
1 12
2 15
3 14
4 19
5 23
6 25
7 29

#check type of *first_col*print( type (first_col))

<class 'pandas.core.frame.DataFrame'>

Дополнительные ресурсы

Как добавить столбец в Pandas DataFrame
Как вставить столбец в фрейм данных Pandas
Как создать новый столбец на основе условия в Pandas