Python Pandas: indexación y selección de datos
En este capítulo, discutiremos cómo dividir y cortar la fecha y, en general, obtener el subconjunto del objeto pandas.
Los operadores de indexación de Python y NumPy "[]" y el operador de atributo "." proporcionan un acceso rápido y sencillo a las estructuras de datos de Pandas en una amplia gama de casos de uso. Sin embargo, dado que el tipo de datos a los que se accede no se conoce de antemano, el uso directo de operadores estándar tiene algunos límites de optimización. Para el código de producción, le recomendamos que aproveche los métodos optimizados de acceso a datos de pandas que se explican en este capítulo.
Pandas ahora admite tres tipos de indexación de ejes múltiples; los tres tipos se mencionan en la siguiente tabla:
| No Señor | Indexación y descripción |
|---|---|
| 1 | .loc() Basado en etiquetas |
| 2 | .iloc() Basado en enteros |
| 3 | .ix() Basado en etiquetas y enteros |
.loc ()
Los pandas proporcionan varios métodos para tener puramente label based indexing. Al cortar, también se incluye el límite inicial. Los números enteros son etiquetas válidas, pero se refieren a la etiqueta y no a la posición.
.loc() tiene múltiples métodos de acceso como -
- Una sola etiqueta escalar
- Una lista de etiquetas
- Un objeto de corte
- Una matriz booleana
loctoma dos operadores individuales / lista / rango separados por ','. El primero indica la fila y el segundo indica las columnas.
Ejemplo 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']Sus output es como sigue -
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: float64Ejemplo 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']]Sus output es como sigue -
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.621420Ejemplo 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']]Sus output es como sigue -
A C
a 0.391548 0.745623
b -0.070649 1.620406
f 0.613709 0.286414
h 1.122680 -1.621420Ejemplo 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']Sus output es como sigue -
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.741470Ejemplo 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']>0Sus output es como sigue -
A False
B True
C False
D False
Name: a, dtype: bool.iloc ()
Pandas proporciona varios métodos para obtener una indexación basada exclusivamente en números enteros. Como Python y Numpy, estos son0-based indexación.
Los diversos métodos de acceso son los siguientes:
- Un entero
- Una lista de enteros
- Un rango de valores
Ejemplo 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]Sus output es como sigue -
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.026251Ejemplo 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]Sus output es como sigue -
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.130568Ejemplo 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]Sus output es como sigue -
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 ()
Además de puramente basado en etiquetas y basado en números enteros, Pandas proporciona un método híbrido para selecciones y subconjuntos del objeto usando el operador .ix ().
Ejemplo 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]Sus output es como sigue -
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.026251Ejemplo 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']Sus output es como sigue -
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 notaciones
Obtener valores del objeto Pandas con indexación de ejes múltiples utiliza la siguiente notación:
| Objeto | Indexadores | Tipo de retorno |
|---|---|---|
| Serie | s.loc [indexador] | Valor escalar |
| Marco de datos | df.loc [row_index, col_index] | Objeto de serie |
| Panel | p.loc [item_index, major_index, minor_index] | p.loc [item_index, major_index, minor_index] |
Note − .iloc() & .ix() aplica las mismas opciones de indexación y valor de retorno.
Veamos ahora cómo se puede realizar cada operación en el objeto DataFrame. Usaremos el operador de indexación básico '[]' -
Ejemplo 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']Sus output es como sigue -
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 pasar una lista de valores a [] para seleccionar esas columnas.
Ejemplo 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']]Sus output es como sigue -
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.372696Ejemplo 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]Sus output es como sigue -
Columns: [A, B, C, D]
Index: []Acceso a atributos
Las columnas se pueden seleccionar utilizando el operador de atributo '.'.
Ejemplo
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(8, 4), columns = ['A', 'B', 'C', 'D'])
print df.ASus output es como sigue -
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