python - matplotlib 中用虚线代替缺失值

Question

我有一些数据的数组，其中一些值丢失了

y = np.array([np.NAN, 45, 23, np.NAN, 5, 14, 22, np.NAN, np.NAN, 18, 23])

当我绘制它时，我缺少这些 NAN(这是预期的)

fig, ax = plt.subplots()
ax.plot(y)
plt.show()

我想要的是一条连接缺失片段的虚线。例如，如果缺少 3 的数据点，则应该有一条虚线连接 2 和 4 之间的现有点(对于缺少数据点 7 和 8 也是如此。如果数据点位于区间边缘(数据点 0)，则 I希望有一条水平线连接它们(想象上一个/下一个数据点与可用边缘相同)。

<小时/>

问题我saw这里询问如何删除这些空段(不是我想要的)。我可以通过创建另一个数组来解决这个问题，该数组将插入缺失值和所有其他值 NAN，但对我来说它看起来很复杂。

因为这看起来很常见，所以我希望有一种更简单的方法。

Answer 1

我想说链接问题的解决方案可以直接应用在这里，在直线后面绘制一条虚线。

import numpy as np
import matplotlib.pyplot as plt

y = np.array([np.NAN, 45, 23, np.NAN, 5, 14, 22, np.NAN, np.NAN, 18, 23])
x = np.arange(0, len(y))
mask = np.isfinite(y)

fig, ax = plt.subplots()
line, = ax.plot(x[mask],y[mask], ls="--",lw=1)
ax.plot(x,y, color=line.get_color(), lw=1.5)

plt.show()

为了在边缘值的情况下考虑水平线，可以检查它们是否为 nan 并将其替换为相邻值。

import numpy as np
import matplotlib.pyplot as plt

y = np.array([np.NAN, 45, 23, np.NAN, 5, 14, 22, np.NAN, np.NAN, 18, 23,np.NAN])
x = np.arange(0, len(y))
yp = np.copy(y)
if ~np.isfinite(y[0]): yp[0] = yp[1]
if ~np.isfinite(y[-1]): yp[-1] = yp[-2]

mask = np.isfinite(yp)


fig, ax = plt.subplots()
line, = ax.plot(x[mask],yp[mask], ls="--",lw=1)
ax.plot(x,y, color=line.get_color(), lw=1.5)

plt.show()