Matplotlib - соединительные линии с использованием методов ma.masked.where

Question

Так что я использую методы numpy.ma.masked для построения линий при определенных условиях, но я хотел бы соединить все последовательные строки. Например, с помощью этого кода:

import pylab as plt 
import numpy as np 
x = np.linspace(0,10,100) 
y = -1.0 + 0.2*x 
plt.plot(x,np.ma.masked_greater_equal(y,0)) 
plt.plot(x,np.ma.masked_less_equal(y,0),'r') 

я получаю следующий результат: Так что это умный способ подключения линий поэтому имеет непрерывную линию, которая меняет свой цвет?

Answer 1

Посмотрите на свои значения y. Это выглядит следующим образом:

array([-1.  , -0.97979798, -0.95959596, -0.93939394, -0.91919192, 
     -0.8989899 , -0.87878788, -0.85858586, -0.83838384, -0.81818182, 
     -0.7979798 , -0.77777778, -0.75757576, -0.73737374, -0.71717172, 
     -0.6969697 , -0.67676768, -0.65656566, -0.63636364, -0.61616162, 
     -0.5959596 , -0.57575758, -0.55555556, -0.53535354, -0.51515152, 
     -0.49494949, -0.47474747, -0.45454545, -0.43434343, -0.41414141, 
     -0.39393939, -0.37373737, -0.35353535, -0.33333333, -0.31313131, 
     -0.29292929, -0.27272727, -0.25252525, -0.23232323, -0.21212121, 
     -0.19191919, -0.17171717, -0.15151515, -0.13131313, -0.11111111, 
     -0.09090909, -0.07070707, -0.05050505, -0.03030303, -0.01010101, 
     0.01010101, 0.03030303, 0.05050505, 0.07070707, 0.09090909, 
     0.11111111, 0.13131313, 0.15151515, 0.17171717, 0.19191919, 
     0.21212121, 0.23232323, 0.25252525, 0.27272727, 0.29292929, 
     0.31313131, 0.33333333, 0.35353535, 0.37373737, 0.39393939, 
     0.41414141, 0.43434343, 0.45454545, 0.47474747, 0.49494949, 
     0.51515152, 0.53535354, 0.55555556, 0.57575758, 0.5959596 , 
     0.61616162, 0.63636364, 0.65656566, 0.67676768, 0.6969697 , 
     0.71717172, 0.73737374, 0.75757576, 0.77777778, 0.7979798 , 
     0.81818182, 0.83838384, 0.85858586, 0.87878788, 0.8989899 , 
     0.91919192, 0.93939394, 0.95959596, 0.97979798, 1.  ]) 

Вы заметите, что нет никакого значения 0.0, таким образом, две линии никогда не будут трогать.

Вы можете решить эту проблему, добавив еще одно значение для вашего x массива (т.е. 101 значений, поэтому у вас есть интервал 0.1, вместо 0.10101. Кроме того, необходимо удалить _equal из масок, в противном случае они не будут трогать (вы в настоящее время маскировать значение в y=0. в обоих случаях).

import pylab as plt 
import numpy as np 
x = np.linspace(0.,10.,101) 
y = -1.0 + 0.2*x 
plt.plot(x,np.ma.masked_greater(y,0.)) 
plt.plot(x,np.ma.masked_less(y,0.),'r')