Pandas loc vs iloc: qual a diferença?
Quando se trata de selecionar linhas e colunas de um DataFrame do pandas, loc e iloc são duas funções comumente usadas.
Aqui está a diferença sutil entre as duas funções:
- loc seleciona linhas e colunas com rótulos específicos
- iloc seleciona linhas e colunas em posições inteiras específicas
Os exemplos a seguir mostram como usar cada função na prática.
Exemplo 1: Como usar loc no Pandas
Suponha que temos o seguinte DataFrame do pandas:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'], ' points ': [5, 7, 7, 9, 12, 9, 9, 4], ' assists ': [11, 8, 10, 6, 6, 5, 9, 12]}, index=['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H']) #view DataFrame df team points assists A A 5 11 B A 7 8 C A 7 10 D A 9 6 E B 12 6 F B 9 5 G B 9 9 H B 4 12
Podemos usar loc para selecionar linhas específicas do DataFrame com base em seus rótulos de índice:
#select rows with index labels 'E' and 'F'
df. loc [[' E ',' F ']]
team points assists
E B 12 6
F B 9 5
Podemos usar loc para selecionar linhas e colunas específicas do DataFrame com base em seus rótulos:
#select 'E' and 'F' rows and 'team' and 'assists' columns
df. loc [[' E ', ' F '], [' team ', ' assists ']]
team assists
E B 12
F B 9
Podemos usar loc com o argumento : para selecionar intervalos de linhas e colunas com base em seus rótulos:
#select 'E' and 'F' rows and 'team' and 'assists' columns
df. loc [' E ':,:' assists ']
team points assists
E B 12 6
F B 9 5
G B 9 9
H B 4 12
Exemplo 2: Como usar o iloc no Pandas
Suponha que temos o seguinte DataFrame do pandas:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'], ' points ': [5, 7, 7, 9, 12, 9, 9, 4], ' assists ': [11, 8, 10, 6, 6, 5, 9, 12]}, index=['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H']) #view DataFrame df team points assists A A 5 11 B A 7 8 C A 7 10 D A 9 6 E B 12 6 F B 9 5 G B 9 9 H B 4 12
Podemos usar iloc para selecionar linhas específicas do DataFrame com base em sua posição inteira:
#select rows in index positions 4 through 6 (not including 6)
df. iloc [4:6]
team points assists
E B 12 6
F B 9 5
Podemos usar iloc para selecionar linhas e colunas específicas do DataFrame com base em suas posições de índice:
#select rows in range 4 through 6 and columns in range 0 through 2
df. iloc [4:6, 0:2]
team assists
E B 12
F B 9
Podemos usar loc com o argumento : para selecionar intervalos de linhas e colunas com base em seus rótulos:
#select rows from 4 through end of rows and columns up to third column
df. iloc [4: , :3]
team points assists
E B 12 6
F B 9 5
G B 9 9
H B 4 12
Recursos adicionais
Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:
Como selecionar linhas com base em múltiplas condições usando Pandas Loc
Como selecionar linhas com base nos valores das colunas no Pandas
Como selecionar linhas por índice no Pandas