Вы можете использовать следующий синтаксис, чтобы получить первый столбец кадра данных 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