Obtenha a primeira linha dos Pandas DataFrame de coluna fornecida

Suraj Joshi 30 janeiro 2023
  1. Obtenha a primeira linha de uma coluna específica no DataFrame usando Series.loc()
  2. Obtenha a primeira linha de uma coluna específica no DataFrame usando Series.loc()
Obtenha a primeira linha dos Pandas DataFrame de coluna fornecida

Este tutorial explica como podemos obter a primeira linha de uma determinada coluna no DataFrame usando o método Series.loc() e Series.iloc().

Usaremos o exemplo de DataFrame abaixo neste artigo.

import pandas as pd

roll_no = [501, 502, 503, 504, 505]

student_df = pd.DataFrame(
    {
        "Name": ["Jennifer", "Travis", "Bob", "Emma", "Luna", "Anish"],
        "Gender": ["Female", "Male", "Male", "Female", "Female", "Male"],
        "Age": [17, 18, 17, 16, 18, 16],
    },
    index=roll_no,
)

print(student_df)

Resultado:

         Name Gender Age
501 Jennifer Female   17
502    Travis    Male   18
503       Bob    Male   17
504      Emma Female   16
505      Luna Female   18
506     Anish    Male   16

Obtenha a primeira linha de uma coluna específica no DataFrame usando Series.loc()

Para obter uma linha específica de um objeto Series usando Series.loc(), simplesmente passamos o nome do índice da linha como um argumento para o método Series.loc().

Cada coluna do DataFrame é um objeto Series, e podemos usar o método .loc() para selecionar qualquer entrada da coluna fornecida.

import pandas as pd

roll_no = [501, 502, 503, 504, 505, 506]

student_df = pd.DataFrame(
    {
        "Name": ["Jennifer", "Travis", "Bob", "Emma", "Luna", "Anish"],
        "Gender": ["Female", "Male", "Male", "Female", "Female", "Male"],
        "Age": [17, 18, 17, 16, 18, 16],
    },
    index=roll_no,
)

print("The DataFrame is:")
print(student_df, "\n")

first_row = student_df["Name"].loc[501]

print("First row from Name column is:")
print(first_row)

Resultado:

The DataFrame is:
         Name Gender Age
501 Jennifer Female   17
502    Travis    Male   18
503       Bob    Male   17
504      Emma Female   16
505      Luna Female   18
506     Anish    Male   16

First row from Name column is:
Jennifer

Ele seleciona a primeira linha da coluna Name do DataFrame student_df e a imprime. Passamos o índice da primeira linha, ou seja, 501 para selecionar a primeira linha.

Alternativamente, podemos passar o índice da primeira linha e o nome da coluna especificada como argumentos para o método loc() para extrair a entrada na primeira linha da coluna especificada no DataFrame.

import pandas as pd

roll_no = [501, 502, 503, 504, 505, 506]

student_df = pd.DataFrame(
    {
        "Name": ["Jennifer", "Travis", "Bob", "Emma", "Luna", "Anish"],
        "Gender": ["Female", "Male", "Male", "Female", "Female", "Male"],
        "Age": [17, 18, 17, 16, 18, 16],
    },
    index=roll_no,
)

print("The DataFrame is:")
print(student_df, "\n")

first_name = student_df.loc[501, "Name"]

print("First row from Name column is:")
print(first_name)

Resultado:

The DataFrame is:
         Name Gender Age
501 Jennifer Female   17
502    Travis    Male   18
503       Bob    Male   17
504      Emma Female   16
505      Luna Female   18
506     Anish    Male   16

First row from Name column is:
Jennifer

Ele seleciona o valor da coluna Name e a primeira linha com valor de índice 503.

Obtenha a primeira linha de uma coluna específica no DataFrame usando Series.loc()

Para obter uma linha específica do DataFrame usando Series.iloc(), passamos o índice inteiro da linha como um argumento para o método Series.iloc().

import pandas as pd

roll_no = [501, 502, 503, 504, 505, 506]

student_df = pd.DataFrame(
    {
        "Name": ["Jennifer", "Travis", "Bob", "Emma", "Luna", "Anish"],
        "Gender": ["Female", "Male", "Male", "Female", "Female", "Male"],
        "Age": [17, 18, 17, 16, 18, 16],
    },
    index=roll_no,
)

print("The DataFrame is:")
print(student_df, "\n")

first_row = student_df["Name"].iloc[0]

print("First row from Name column is:")
print(first_row)

Resultado:

The DataFrame is:
         Name Gender Age
501 Jennifer Female   17
502    Travis    Male   18
503       Bob    Male   17
504      Emma Female   16
505      Luna Female   18
506     Anish    Male   16

First row from Name column is:
Jennifer

Ele seleciona a primeira linha da coluna Name do DataFrame student_df e a imprime. Passamos o índice inteiro da primeira linha, ou seja, 0, uma vez que o índice começa em 0.

Alternativamente, podemos passar o índice inteiro da primeira linha e o índice da coluna especificada como argumentos para o método iloc() para extrair a entrada na primeira linha da coluna especificada no DataFrame.

import pandas as pd

roll_no = [501, 502, 503, 504, 505, 506]

student_df = pd.DataFrame(
    {
        "Name": ["Jennifer", "Travis", "Bob", "Emma", "Luna", "Anish"],
        "Gender": ["Female", "Male", "Male", "Female", "Female", "Male"],
        "Age": [17, 18, 17, 16, 18, 16],
    },
    index=roll_no,
)

print("The DataFrame is:")
print(student_df, "\n")

first_name = student_df.iloc[0, 0]

print("Name of student at first row is:")
print(first_name)

Resultado:

The DataFrame is:
         Name Gender Age
501 Jennifer Female   17
502    Travis    Male   18
503       Bob    Male   17
504      Emma Female   16
505      Luna Female   18
506     Anish    Male   16

Name of student at first row is:
Jennifer

Ele seleciona o valor da primeira linha e primeira coluna do DataFrame.

Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn

Artigo relacionado - Pandas DataFrame Column