matplotlib NaN 与 pylab NaN

Question

我有两段相似的 matplotlib 代码，它们产生不同的结果。

1：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,10,100)
y = np.linspace(0,10,100)
y[10:40] = np.nan
plt.plot(x,y)
plt.savefig('fig')

2：

from pylab import *

x = linspace(0,10,100)
y = linspace(0,10,100)
y[10:40] = np.nan
plot(x,y)
savefig('fig')

代码 #1 生成一条直线，其中 NaN 区域用不同颜色的实线填充

代码 #2 生成一个带有直线的图形，但没有用直线填充 NaN 区域。相反，那里有一个差距。

我怎样才能让代码#1 产生一个间隙来代替NaN 的代码#2。我已经用谷歌搜索了几天，但一无所获。任何帮助或建议将不胜感激。提前致谢

Answer 1

只是为了解释可能发生的事情：

您显示的两段代码是相同的。如果它们自己调用，它们将始终产生相同的输出。 pylab 基本上只需要几行代码即可：（除此之外还有更多内容，但这是基本思想。）

from numpy import *
from matplotlib.mlab import *
from matplotlib.pyplot import *

pylab.plot 绝对无法引用与plt.plot 不同的函数

---

但是，如果您只是调用plt.plot（或pylab.plot，它们是同一个函数），它会绘制在当前图上。

如果您之前在该图上绘制了一些东西，它仍然会在那里。 （如果你熟悉 matlab，matplotlib 默认为hold('on')。你可以用plt.hold 更改它，但最好在 python 中更明确，然后创建一个新图形。）

基本上，您可能是这样做的：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,10,100)
y = np.linspace(0,10,100)

plt.plot(x,y)
plt.savefig('fig')

然后，在同一个交互式 ipython 会话中，您执行了以下操作：

y[10:40] = np.nan
plt.plot(x, y)
plt.savefig('fig')

因为你没有打电话给show，所以现在的数字还是和以前一样。 “完整”线仍然存在于第二条线下方，并且带有 NaN 的第二条线是不同的颜色，因为您绘制在相同的轴上。

这是使用面向对象接口是个好主意的众多原因之一。这样您就可以准确地了解您正在绘制的坐标轴和图形。

例如：

fig, ax = plt.subplots()
ax.plot(x, y)
fig.savefig('test.png')

如果您不打算这样做，至少在您想要一个新图形时总是明确地创建一个新图形和/或轴。 （例如，首先调用plt.figure()）