如何在同一张图中绘制 contourf 和我的图形

Question

我有一个显示 contourf 图的图和另一个显示我之前制作的图的图，我想在同一个图上绘制这两个图，我该怎么办？这是我的 contourf 图的代码：

import pylab as pl
from pylab import *
import xlrd
import math
import itertools
from matplotlib import collections as mc
import matplotlib.pyplot as plt
import copy as dc
import pyexcel
from pyexcel.ext import xlsx
import decimal

x_list = linspace(0, 99, 100)
y_list = linspace(0, 99, 100)
X, Y = meshgrid(x_list, y_list, indexing='xy')

Z = [[0 for x in range(len(x_list))] for x in range(len(y_list))]
for each_axes in range(len(Z)):
    for each_point in range(len(Z[each_axes])):
        Z[len(Z)-1-each_axes][each_point] = power_at_each_point(each_point, each_axes)

figure()
CP2 = contourf(X, Y, Z, cmap=plt.get_cmap('Reds'))
colorbar(CP2)
title('Coverage Plot')
xlabel('x (m)')
ylabel('y (m)')
show()

这是我之前绘制的剧情代码：

lc = mc.LineCollection(lines, linewidths=3)
fig, ax = pl.subplots()
ax.add_collection(lc)
ax.autoscale()
ax.margins(0.05)

#The code blow is just for drawing the final plot of the building.
Nodes = xlrd.open_workbook(Node_file_location)
sheet = Nodes.sheet_by_index(0)
Node_Order_Counter = range(1, sheet.nrows + 1)
In_Node_Order_Counter = 0
for counter in range(len(Node_Positions_Ascending)):
    plt.plot(Node_Positions_Ascending[counter][0],     Node_Positions_Ascending[counter][1], marker='o', color='r',
             markersize=6)
    pl.text(Node_Positions_Ascending[counter][0],     Node_Positions_Ascending[counter][1],
            str(Node_Order_Counter[In_Node_Order_Counter]),
            color="black", fontsize=15)
    In_Node_Order_Counter += 1
#Plotting the different node positions on our plot & numbering them
pl.show()

Answer 1

如果没有您的数据，我们无法看到剧情应该是什么样子，但我有一些一般性建议。

不要使用 pylab。如果你绝对必须使用它，请在它的命名空间内使用它，不要做 from pylab import *。这使得代码非常草率 - 例如，linspace 和 meshgrid 实际上来自 numpy，但是当你使用 pylab 时很难说出来。
对于复杂的绘图，甚至不要使用 pyplot。相反，使用直接对象绘图界面。例如，要在等高线图之上制作法线图（如您所愿），您可以执行以下操作：

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

x = np.linspace(1, 5, 20)
y = np.linspace(2, 5, 20)
z = x[:,np.newaxis] * (y[np.newaxis,:])**2

xx, yy = np.meshgrid(x, y)

ax.contourf(xx, yy, z, cmap='Reds')
ax.plot(x, 0.2*y**2)

plt.show()

请注意，我只使用 pyplot 创建图形和坐标轴，并显示它们。实际绘图是使用 AxesSubplot 对象完成的。

如何在同一张图中绘制 contourf 和我的图形

How to plot contourf and my graph in the same figure

python

plot

matplotlib

figure

contourf