0

我有一个显示轮廓图的图和另一个显示我之前制作的图,我想在同一个图上绘制两者我应该怎么做?这是我的轮廓图的代码:

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()
4

1 回答 1

1

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

  1. 不要使用 pylab。如果您绝对必须使用它,请在其命名空间中使用它,并且不要使用from pylab import *. 它使得代码非常草率——例如,linspace 和 meshgrid 实际上来自 numpy,但是当你使用 pylab 时很难说出来。
  2. 对于复杂的绘图,甚至不要使用 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 对象完成的。

于 2015-05-09T18:26:37.120 回答