matplotlib: Traza matrices numpy con Ninguno como valores
Tengo una matriz que se parece a:
k = numpy.array([(1.,0.001), (1.1, 0.002), (None, None),
(1.2, 0.003), (0.99, 0.004)])
Quiero trazar los valores que no están(None, None)
y mantener el índice del valor de la matriz. Es decir, quiero un hueco donde quiera que haya un(None, None)
valor.
Cuando se haga eso me gustaría trazar
y = k[:,0] + k[:,1]
pero ni siquiera puedo agregar las matrices juntas. Intenté enmascarar la matriz, pero perdí los valores de índice del originalk
formación.
Un ejemplo mínimo:
import matplotlib.pyplot as pyplot
import numpy
x = range(5)
k = numpy.array([(1.,0.001), (1.1, 0.002), (None, None),
(1.2, 0.003), (0.99, 0.004)])
Fig, ax = pyplot.subplots()
# This plots a gap---as desired
ax.plot(x, k[:,0], 'k-')
# I'd like to plot
# k[:,0] + k[:,1]
# but I can't add None
# Here I get rid of the (None, None) values so I can add
# But I lose the original indexing
mask = k != (None, None)
y = k[mask].reshape((-1,2))
ax.plot(range(len(y)), y[:,0]+y[:,1], 'k--')