Con éste Notebook trataremos de dar a conocer plotly
, un servicio web gratuito para generar gráficos interactivos y que permite la creación de proyectos colaborativos.
Éste Notebook emplea los siguentes paquetes:
pandas
0.14.0, para importar y analizar estructuras de datos en Python.matplotlib
1.3.1, para generar diversisos tipos de gráficos con calidad de imprenta.plotly
1.1.2, para generar gráficos interactivos para la web.que iremos cargando conforme los vayamos a utilizar.
Empezaremos por importar los datos con pandas
. Podeis encontrar una introducción a pandas
en éste post de Pybonacci.
La estructura de datos con la que vamos a trajar es un CSV
con datos de telemetría correspondientes a un BMW Z4 GT3 compitiendo en el circuito Brands Hatch Indy.
El paquete de datos nos lo ha proporcionado Steve Barker, un mecánico que pasa la mayor parte de su tiempo en la GP2. Pero aquí sólo vamos a trabajar con los datos de la vuelta 27 de la primera carrera.
El propio pandas
incluye rutinas para importar datos desde diversas fuentes, ya sea un simple CSV
o un Excel de multiples hojas.
In [1]:
# Importamos pandas de la manera habitual
import pandas as pd
Pero antes de proceder a cargar los datos debemos conocer un poco la estructura del archivo en cuestión del que he recortado una muestra.
Format;MoTeC CSV File;;;Workbook;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Venue;brandshatch indy;;;Worksheet;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Vehicle;bmwz4gt3;;;Vehicle Desc;bmwz4gt3;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Driver;Steve Barker;;;Engine ID;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Device;ADL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Comment;;;;Session;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Log Date;02/07/2014;;;Origin Time;1196.94;s;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Log Time;23:05:43;;;Start Time;1196.95;s;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Sample Rate;60;Hz;;End Time;1240.017;s;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Duration;43.067;s;;Start Distance;49849;m;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Range;Lap 27;;;End Distance;51765;m;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
Time;Distance;Gear;Engine RPM;Manifold Pres;Throttle Pos;Brake Pedal Pos;Clutch
s;m;;rpm;kpa;%;%;%;deg;N.m;;m;km/h;G;G;G;deg;deg;deg;deg/s;deg/s;deg/s;C;kPa;V;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
0;0;5;8657;0.99752352;100;0;100;-5.724613;1.5780677;26;0.6267;225.407808;-0.296
0.017;1;5;8658;0.9975208;100;0;100;-6.10625;1.7365016;26;1.66957;225.514112;-0.
0.033;2;5;8654;0.99751824;100;0;100;-6.160785;1.929843;26;2.71285;225.58688;-0.
0.05;3;5;8670;0.99751552;100;0;100;-6.160785;2.5375902;26;3.7565;225.653952;-0.
Como bien podemos apreciar, las primeras 11 líneas aportan información generada por MoTeC al exportar los datos. Las cabeceras de lo que realmente nos interesa son las líneas 14 y 15, siendo la primera la descripción del parámetro y la segunda las unidades de medida. Y ya en la línea 18 empiezan los datos.
Tomaremos como índice del DataFrame
la variable tiempo (aunque también podríamos optar por la distancia).
In [2]:
# Cargamos el archivo. Recordemos que pandas, y Python en general, es 0-indexed.
data = pd.io.parsers.read_csv('brandshatch_lap27.csv', sep=';', header=13, index_col=0, skiprows=[14,15,16])
# Otra forma alternativa de leer los datos de un archivo CSV sería a través del
# propio DataFrame de pandas, pero éste no nos permite saltarnos filas de datos:
# data = pd.DataFrame.from_csv('brandshatch_lap27.csv', sep=';', header=13, index_col=0)
data.head(5)
Out[2]:
Podemos ver que el archivo es inmenso. Contiene 215 con diferentes parámetros como revoluciones del motor, posición del pedal del acelarador, velocidad, etc. Un sinfín de datos que marearían al más experto de los ingenieros de pista.
De todos modos, aquí no vamos a analizar en detalle la vuelta realizada por Steve Barker, sino que nos centraremos exclusivamente en visualizar los datos.
matplotlib
matplotlib
es el paquete más empleado en Python para generar diversos tipos de gráficos de alta calidad.
Y es precisamente lo que utiliza pandas
para generar sus propias gráficas como veremos a continuación.
Empezamos por importar matplotlib
y, de paso, crear un mapa de colores personalizado más vistoso que el que viene por defecto.
La idea la he tomado de Randal S. Olson.
In [3]:
import matplotlib.pyplot as plt
import matplotlib as mpl
%matplotlib inline
In [4]:
# Color map
tableau20 = [(31, 119, 180), (174, 199, 232), (255, 127, 14), (255, 187, 120),
(44, 160, 44), (152, 223, 138), (214, 39, 40), (255, 152, 150),
(148, 103, 189), (197, 176, 213), (140, 86, 75), (196, 156, 148),
(227, 119, 194), (247, 182, 210), (127, 127, 127), (199, 199, 199),
(188, 189, 34), (219, 219, 141), (23, 190, 207), (158, 218, 229)]
tableau10 = [tableau20[i*2] for i in range(10)]
# Scale the RGB values to the [0, 1] range, which is the format matplotlib accepts.
for i in range(len(tableau20)):
r, g, b = tableau20[i]
tableau20[i] = (r / 255., g / 255., b / 255.)
for i in range(len(tableau10)):
r, g, b = tableau10[i]
tableau10[i] = (r / 255., g / 255., b / 255.)
# Set the default color cycle
mpl.rcParams['axes.color_cycle'] = tableau10
In [5]:
fig, ax = plt.subplots(3, sharex=True, figsize=(10,6))
# Revoluciones del motor
data['Engine RPM'].plot(ax=ax[0], linewidth=2)
ax[0].set_xlabel('')
ax[0].set_ylabel('Engine RPM')
ax[0].legend(loc='lower right')
# Velocidad del vehículo
data['Ground Speed'].plot(ax=ax[1], linewidth=2)
ax[1].set_xlabel('')
ax[1].set_ylabel('Speed (kph)')
ax[1].legend(loc='lower right')
# Pedales
data['Throttle Pos'].plot(ax=ax[2], linewidth=2)
data['Brake Pedal Pos'].plot(ax=ax[2], linewidth=2)
ax[2].set_xlabel('Time (s)')
ax[2].set_ylabel('Position (%)')
ax[2].set_ylim(-10,105)
ax[2].legend(loc='center right')
for j in range(3):
ax[j].set_axisbelow(True)
# Quitamos los bordes superior y derecho
ax[j].spines["top"].set_visible(False)
ax[j].spines["right"].set_visible(False)
# Dejamos sólo los ticks abajo y a la izquierda
ax[j].get_xaxis().tick_bottom()
ax[j].get_yaxis().tick_left()
Lo que hemos realizado en la gráfica de arriba son tres subgráficas que comparten el mismo eje x. De ese modo podemos ver la relación entre los diversos parámetros.
Otro tipo de gráfico que nos puede dar mucha información es la relación entre la velocidad del vehículo y las revoluciones del motor.
Para ello podemos utilizar en pandas
el tipo scatter
y colorear los valores en función de la marcha engranada.
Eso lo conseguimos con la herramientas groupby
de pandas
.
In [6]:
fig1, ax1 = plt.subplots(figsize=(10,6))
for key, grp in data.groupby(['Gear']):
grp.plot('Ground Speed','Engine RPM',kind='scatter', ax=ax1, label=key, color=tableau10[key])
ax1.set_xlabel('Ground Speed (kph)')
ax1.set_axisbelow(True)
ax1.legend(title='Gear', loc='upper left')
# Quitamos los márgenes derecho y superior
ax1.spines["top"].set_visible(False)
ax1.spines["right"].set_visible(False)
# Dejamos sólo los ticks abajo y a la izquierda
ax1.get_xaxis().tick_bottom()
ax1.get_yaxis().tick_left()
In [7]:
fig2, ax2 = plt.subplots(figsize=(10,6))
data['Throttle Pos'].hist(ax=ax2)
ax2.set_xlabel('Throttle position (%)')
ax2.set_xlim(0,100)
ax2.set_ylabel('Time spent (@ 60 Hz)')
ax2.set_axisbelow(True)
# Quitamos los márgenes derecho y superior
ax2.spines["top"].set_visible(False)
ax2.spines["right"].set_visible(False)
# Dejamos sólo los ticks abajo y a la izquierda
ax2.get_xaxis().tick_bottom()
ax2.get_yaxis().tick_left()
plotly
plotly
ofrece un API excelente para crear gráficas interactivas que pueden ser incluidas en webs y blogs.
El paquete no viene incluido en los repositorios de Continuum Analytics (conda
) pero se puede instalar de manera muy sencilla con pip
:
pip install plotly
In [8]:
import plotly.plotly as py
from plotly.graph_objs import *
Para poder utilizar plotly
necesitamos una credenciales. Registrarse es gratuito, y ofrecen almacenamiento ilimitado para gráficos públicos. Sin embargo, el número de archivos privados que podemos alojar es limitado en la versión gratuita.
Para mayor comodidad podemos guardar nuestras credenciales en el sistema de modo siguiente,
>>> import plotly.tools as tls
>>> tls.set_credentials_file(username="your_username",
api_key="your_api_key")
que rellenaremos con nuestros datos, que encontraremos en la sección de Configuration
de nuestra cuenta.
Nuestras credenciales las podremos recuperar, a partir de entonces, de la siguente manera.
In [9]:
my_creds = py.get_credentials()
username = my_creds['username']
api_key = my_creds['api_key']
py.sign_in(username, api_key)
Podemos utilizar plotly
de dos maneras.
La primera de ellas es generar las gráficas directamente con la API.
La segunda, y la que resultará más cómoda para la mayoría, es convertir directamente las figuras generadas con matplotlib
.
In [10]:
# Publicar una figura de matplotlib en la web
unique_url = py.iplot_mpl(fig, filename='pybonacci/velocidades', strip_style=True)
In [11]:
unique_url1 = py.iplot_mpl(fig1, filename='pybonacci/rpm', strip_style=True)
In [12]:
unique_url2 = py.iplot_mpl(fig2, filename='pybonacci/acelerador', strip_style=True)
plotly
Como podemos ver en las gráficas superiores, la conversión a plotly
todavía no es perfecta. El conversor interpreta las leyendas de matplotlib
como anotaciones de texto. Por suerte, eso lo podemos subsanar fácilmente.
Necesitaremos importar una serie de paquetes adicionales, disponibles en las versiones >1 de plotly
.
In [13]:
# Herramientas Python/Plotly
import plotly.tools as tls
# Objetos para componer Leyendas
from plotly.graph_objs import Legend
Convertimos la figura de matplotlib
a una figura de plotly
,
In [14]:
py_fig = tls.mpl_to_plotly(fig, strip_style=True)
y eliminamos todas las anotaciones de texto y añadimos una leyenda en la esquina superior derecha.
In [15]:
# Borrar anotaciones
py_fig['layout'].pop('annotations',None)
# Añadir leyenda
py_fig['layout'].update(dict(showlegend=True))
Ya podemos publicar la nueva gráfica, ésta vez sólo con py.plot()
o py.iplot()
pues ya hemos hecho la conversión a plotly
unos pasos atrás.
In [16]:
unique_url3 = py.iplot(py_fig, filename='pybonacci/velocidades_edit')
Repetimos el procedimiento para la segunda gráfica, pero en esta ocasión colocaremos la leyenda arriba a la izquierda y cambiamos el hovermode
a closest
.
Con hovermode
controlamos cómo se comporta plotly
al pasar el ratón por encima de la gráfica. En modo closest
nos resaltará el valor más cercano al ratón, y en modo compare
, para cada x nos mostrará los diferentes valores que toma la variable.
In [17]:
py_fig1 = tls.mpl_to_plotly(fig1, strip_style=True)
# Borrar anotaciones
py_fig1['layout'].pop('annotations',None)
# Añadir leyenda
py_fig1['layout'].update(dict(showlegend=True,
legend=Legend(x=0,y=1),
hovermode='closest'))
Pero esta vez tenemos que editar los nombres de los puntos que se muestran en la leyenda.
In [18]:
# Editar leyenda
n=0
for key, grp in data.groupby(['Gear']):
py_fig1['data'][n].update(dict(name='Gear {}'.format(key)))
n += 1
In [19]:
unique_url4 = py.iplot(py_fig1, filename='pybonacci/rpm_edit')
Por último nos falta retocar el histograma de posiciones del pedal del acelerador. En ésta gráfica no quedan anotaciones, pero tampoco necesitamos mostrar la leyenda. Aprovechamos también para hacer las barras más anchar reduciendo el bargap
.
In [20]:
py_fig2 = tls.mpl_to_plotly(fig2, strip_style=True)
# Borrar leyenda
py_fig2['layout'].update(dict(showlegend=False,
bargap=0.01))
In [21]:
unique_url5 = py.iplot(py_fig2, filename='pybonacci/acelerador_edit')