Obtención de valores únicos de una columna con Pandas

En muchas ocasiones al trabajar con un DataFrame de Pandas puede que una de las columnas contenta los valores de una etiqueta. La que posiblemente tendrá muchos registros repetidos. Por lo que puede ser interesante obtener el listado de valores únicos de la columna. Afortunadamente los objetos DataFrame de la librería Pandas cuentan con un método para obtener estos. A continuación, se explicarán diferentes ejemplos para la obtención de valores únicos de una columna con Pandas.

A modo de ejemplo en esta entrada se utilizará el conjunto de datos de exoplanetas que se puede encontrar en la librería Seaborn. Para importarlo se ha de importar la librería y utilizas la función load_dataset() como se muestra en el siguiente ejemplo

import seaborn as sb

planets = sb.load_dataset('planets')
planets.head()

            method  number  orbital_period   mass  distance  year
0  Radial Velocity       1         269.300   7.10     77.40  2006
1  Radial Velocity       1         874.774   2.21     56.95  2008
2  Radial Velocity       1         763.000   2.60     19.84  2011
3  Radial Velocity       1         326.030  19.40    110.62  2007
4  Radial Velocity       1         516.220  10.50    119.47  2009

Obtención de valores únicos de una columna

En el conjunto de datos de exoplanetas existe una columna con el método de utilizados para su descubrimiento. Por lo que puede ser interesante obtener el listado de método únicos utilizados. El método de las columnas de Pandas que permite obtener esto valores es unique(). Así el listado de métodos de descubrimiento de planetas se puede obtener aplicando este método a la columna method.

En Analytics Lane

Nuevo test de normalidad interactivo en el laboratorio de Analytics Lane

planets['method'].unique()

array(['Radial Velocity', 'Imaging', 'Eclipse Timing Variations',
       'Transit', 'Astrometry', 'Transit Timing Variations',
       'Orbital Brightness Modulation', 'Microlensing', 'Pulsar Timing',
       'Pulsation Timing Variations'], dtype=object)

En este caso se puede ver que el resultado es un array. En el caso de que desee que el listado de valores sea una lista se puede utilizar el método tolist() para transformarla.

planets['method'].unique().tolist()

['Radial Velocity',
 'Imaging',
 'Eclipse Timing Variations',
 'Transit',
 'Astrometry',
 'Transit Timing Variations',
 'Orbital Brightness Modulation',
 'Microlensing',
 'Pulsar Timing',
 'Pulsation Timing Variations']

El método unique() se puede aplicar también en columnas cuyo contenido sea numérico. Por ejemplo, en la columna con el número.

planets['number'].unique()

array([1, 2, 3, 5, 4, 6, 7])

Número de valores únicos en la columna

Una vez obtenidos el listado con los registros únicos de una columna es trivial obtener el número de estos. Para ello solamente es necesario utilizar la función len() sobre el objeto.

len(planets['method'].unique())

10

Conclusiones

En esta entrada se ha visto cómo obtener los valores únicos de una columna con Pandas. Operación que es necesaria cuando una de las columnas del conjunto de datos contenga un índice. Algo que es bastante habitual.

Imágenes: Pixabay (heecehil)