使用matplotlib在Python上绘制地下水位

Question

我正试图绘制一个数组（一个地下水位）。X轴是我正在绘制的地图的坐标，Y轴对应于该坐标中的地下水位

我正在尝试以下方法：

water_table = flopy.utils.postprocessing.get_water_table(heads_data, nodata=-9999)
x = np.linspace(0, 12000, 200)
plt.plot(x, water_table.T, 'r')

我得到以下图表：

我尝试了散点绘图，这样我就可以添加一条穿过每个点的线，但它不起作用。我不知道是不是因为数组？我怎样才能去掉每个点下面的蓝线，或者添加一些东西使我的图表看起来“更漂亮”

Answer 1

使用numpy，可以过滤掉零：

x = np.linspace(0, 12000, 200)
t = np.array(water_table.T)
plt.plot(x[t>0], t[t>0], 'dodgerblue')

或者，如果您希望在没有数据的情况下有间隙，您可以用NAN替换不需要的值：

x = np.linspace(0, 12000, 200)
t = np.array(water_table.T)
plt.plot(x, np.where(t > 0, t, np.nan), 'navy')