Python Pandas - indexação e seleção de dados
Neste capítulo, discutiremos como fatiar e dividir a data e, em geral, obter o subconjunto do objeto pandas.
Os operadores de indexação Python e NumPy "[]" e o operador de atributo "." fornecem acesso rápido e fácil às estruturas de dados do Pandas em uma ampla variedade de casos de uso. No entanto, como o tipo de dados a ser acessado não é conhecido com antecedência, o uso direto de operadores padrão tem alguns limites de otimização. Para código de produção, recomendamos que você aproveite as vantagens dos métodos otimizados de acesso a dados do pandas explicados neste capítulo.
O Pandas agora oferece suporte a três tipos de indexação de eixos múltiplos; os três tipos são mencionados na tabela a seguir -
| Sr. Não | Indexação e descrição |
|---|---|
| 1 |
.loc() Com base em rótulo |
| 2 |
.iloc() Baseado em inteiro |
| 3 |
.ix() Com base em rótulo e número inteiro |
.loc ()
Pandas fornecem vários métodos para ter puramente label based indexing. Ao fatiar, o limite inicial também é incluído. Os inteiros são rótulos válidos, mas eles se referem ao rótulo e não à posição.
.loc() tem vários métodos de acesso como -
- Um único rótulo escalar
- Uma lista de rótulos
- Um objeto de fatia
- Uma matriz booleana
locleva dois operadores únicos / lista / intervalo separados por ','. O primeiro indica a linha e o segundo indica as colunas.
Exemplo 1
#import the pandas library and aliasing as pd
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4),
index = ['a','b','c','d','e','f','g','h'], columns = ['A', 'B', 'C', 'D'])
#select all rows for a specific column
print df.loc[:,'A']Está output é o seguinte -
a 0.391548
b -0.070649
c -0.317212
d -2.162406
e 2.202797
f 0.613709
g 1.050559
h 1.122680
Name: A, dtype: float64Exemplo 2
# import the pandas library and aliasing as pd
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4),
index = ['a','b','c','d','e','f','g','h'], columns = ['A', 'B', 'C', 'D'])
# Select all rows for multiple columns, say list[]
print df.loc[:,['A','C']]Está output é o seguinte -
A C
a 0.391548 0.745623
b -0.070649 1.620406
c -0.317212 1.448365
d -2.162406 -0.873557
e 2.202797 0.528067
f 0.613709 0.286414
g 1.050559 0.216526
h 1.122680 -1.621420Exemplo 3
# import the pandas library and aliasing as pd
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4),
index = ['a','b','c','d','e','f','g','h'], columns = ['A', 'B', 'C', 'D'])
# Select few rows for multiple columns, say list[]
print df.loc[['a','b','f','h'],['A','C']]Está output é o seguinte -
A C
a 0.391548 0.745623
b -0.070649 1.620406
f 0.613709 0.286414
h 1.122680 -1.621420Exemplo 4
# import the pandas library and aliasing as pd
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4),
index = ['a','b','c','d','e','f','g','h'], columns = ['A', 'B', 'C', 'D'])
# Select range of rows for all columns
print df.loc['a':'h']Está output é o seguinte -
A B C D
a 0.391548 -0.224297 0.745623 0.054301
b -0.070649 -0.880130 1.620406 1.419743
c -0.317212 -1.929698 1.448365 0.616899
d -2.162406 0.614256 -0.873557 1.093958
e 2.202797 -2.315915 0.528067 0.612482
f 0.613709 -0.157674 0.286414 -0.500517
g 1.050559 -2.272099 0.216526 0.928449
h 1.122680 0.324368 -1.621420 -0.741470Exemplo 5
# import the pandas library and aliasing as pd
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4),
index = ['a','b','c','d','e','f','g','h'], columns = ['A', 'B', 'C', 'D'])
# for getting values with a boolean array
print df.loc['a']>0Está output é o seguinte -
A False
B True
C False
D False
Name: a, dtype: bool.iloc ()
O Pandas fornece vários métodos para obter indexação puramente baseada em inteiros. Como python e numpy, estes são0-based indexação.
Os vários métodos de acesso são os seguintes -
- Um inteiro
- Uma lista de inteiros
- Uma gama de valores
Exemplo 1
# import the pandas library and aliasing as pd
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
# select all rows for a specific column
print df.iloc[:4]Está output é o seguinte -
A B C D
0 0.699435 0.256239 -1.270702 -0.645195
1 -0.685354 0.890791 -0.813012 0.631615
2 -0.783192 -0.531378 0.025070 0.230806
3 0.539042 -1.284314 0.826977 -0.026251Exemplo 2
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
# Integer slicing
print df.iloc[:4]
print df.iloc[1:5, 2:4]Está output é o seguinte -
A B C D
0 0.699435 0.256239 -1.270702 -0.645195
1 -0.685354 0.890791 -0.813012 0.631615
2 -0.783192 -0.531378 0.025070 0.230806
3 0.539042 -1.284314 0.826977 -0.026251
C D
1 -0.813012 0.631615
2 0.025070 0.230806
3 0.826977 -0.026251
4 1.423332 1.130568Exemplo 3
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
# Slicing through list of values
print df.iloc[[1, 3, 5], [1, 3]]
print df.iloc[1:3, :]
print df.iloc[:,1:3]Está output é o seguinte -
B D
1 0.890791 0.631615
3 -1.284314 -0.026251
5 -0.512888 -0.518930
A B C D
1 -0.685354 0.890791 -0.813012 0.631615
2 -0.783192 -0.531378 0.025070 0.230806
B C
0 0.256239 -1.270702
1 0.890791 -0.813012
2 -0.531378 0.025070
3 -1.284314 0.826977
4 -0.460729 1.423332
5 -0.512888 0.581409
6 -1.204853 0.098060
7 -0.947857 0.641358.ix ()
Além de baseado em rótulo puro e baseado em inteiro, Pandas fornece um método híbrido para seleções e subconjuntos do objeto usando o operador .ix ().
Exemplo 1
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
# Integer slicing
print df.ix[:4]Está output é o seguinte -
A B C D
0 0.699435 0.256239 -1.270702 -0.645195
1 -0.685354 0.890791 -0.813012 0.631615
2 -0.783192 -0.531378 0.025070 0.230806
3 0.539042 -1.284314 0.826977 -0.026251Exemplo 2
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
# Index slicing
print df.ix[:,'A']Está output é o seguinte -
0 0.699435
1 -0.685354
2 -0.783192
3 0.539042
4 -1.044209
5 -1.415411
6 1.062095
7 0.994204
Name: A, dtype: float64Uso de Notações
Obter valores do objeto Pandas com indexação Multi-eixos usa a seguinte notação -
| Objeto | Indexadores | Tipo de Devolução |
|---|---|---|
| Series | s.loc [indexador] | Valor escalar |
| Quadro de dados | df.loc [row_index, col_index] | Objeto de série |
| Painel | p.loc [item_index, major_index, minor_index] | p.loc [item_index, major_index, minor_index] |
Note − .iloc() & .ix() aplica as mesmas opções de indexação e valor de retorno.
Vamos agora ver como cada operação pode ser executada no objeto DataFrame. Usaremos o operador de indexação básico '[]' -
Exemplo 1
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
print df['A']Está output é o seguinte -
0 -0.478893
1 0.391931
2 0.336825
3 -1.055102
4 -0.165218
5 -0.328641
6 0.567721
7 -0.759399
Name: A, dtype: float64Note - Podemos passar uma lista de valores para [] para selecionar essas colunas.
Exemplo 2
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
print df[['A','B']]Está output é o seguinte -
A B
0 -0.478893 -0.606311
1 0.391931 -0.949025
2 0.336825 0.093717
3 -1.055102 -0.012944
4 -0.165218 1.550310
5 -0.328641 -0.226363
6 0.567721 -0.312585
7 -0.759399 -0.372696Exemplo 3
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
print df[2:2]Está output é o seguinte -
Columns: [A, B, C, D]
Index: []Acesso de Atributo
As colunas podem ser selecionadas usando o operador de atributo '.'.
Exemplo
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
print df.AEstá output é o seguinte -
0 -0.478893
1 0.391931
2 0.336825
3 -1.055102
4 -0.165218
5 -0.328641
6 0.567721
7 -0.759399
Name: A, dtype: float64