Como obter a primeira coluna do pandas dataframe (com exemplos)

Você pode usar a seguinte sintaxe para obter a primeira coluna de um DataFrame do pandas:

 df. iloc [:, 0]

O resultado é uma série de pandas. Se quiser que o resultado seja um DataFrame do pandas, você pode usar a seguinte sintaxe:

 df. iloc [:, :1]

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Obtenha a primeira coluna do Pandas DataFrame (retorne uma série)

O código a seguir mostra como obter a primeira coluna de um DataFrame do 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

O resultado é uma série de pandas:

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

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

Exemplo 2: Obtenha a primeira coluna do Pandas DataFrame (retorne um DataFrame)

O código a seguir mostra como obter a primeira coluna de um DataFrame do pandas e retornar um DataFrame de acordo:

 #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'>

Recursos adicionais

Como adicionar uma coluna a um DataFrame do Pandas
Como inserir uma coluna em um DataFrame do Pandas
Como criar uma nova coluna baseada em uma condição no Pandas