Как получить первый столбец pandas dataframe (с примерами)
Вы можете использовать следующий синтаксис, чтобы получить первый столбец DataFrame pandas:
df. iloc [:, 0]
В результате получилась серия панд. Если вы хотите, чтобы результатом был DataFrame pandas, вы можете использовать следующий синтаксис:
df. iloc [:, :1]
Следующие примеры показывают, как использовать этот синтаксис на практике.
Пример 1. Получить первый столбец из Pandas DataFrame (вернуть серию)
Следующий код показывает, как получить первый столбец DataFrame pandas:
import pandas as pd #createDataFrame 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. Получите первый столбец DataFrame Pandas (верните DataFrame)
Следующий код показывает, как получить первый столбец DataFrame pandas и соответственно вернуть DataFrame:
#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'>
Дополнительные ресурсы
Как добавить столбец в DataFrame Pandas
Как вставить столбец в DataFrame Pandas
Как создать новый столбец на основе условия в Pandas