[英]Images getting merged into single plot while saving plot using matplotlib
[英]Getting plot using matplotlib
def get_plot(x_plot, y_plot, column_span, title, x_axis_1, y_axis_1,
x_axis_2 = None, x_axis_3 = None,
y_axis_2 = None, y_axis_3 = None,
label_1 = None, label_2 = None, label_3 = None):
ax[x_plot][y_plot] = plt.subplot2grid((3, 3), (x_plot, y_plot), colspan = column_span)
if label_1 is None:
ax[x_plot][y_plot].plot(x_axis_1, y_axis_1, linewidth = 2, color = ([0.37, 0.23, 0.37]), marker = 'o')
else:
ax[x_plot][y_plot].plot(x_axis_1, y_axis_1, linewidth = 2, color = ([0.37, 0.23, 0.37]), label = label_1, marker = 'o')
if x_axis_2 is not None and y_axis_2 is not None and label_2 is not None:
ax[x_plot][y_plot].plot(x_axis_2, y_axis_2, linewidth = 2, color = ([0.44, 0.64, 0.69]), label = label_2, marker = 'o')
if title == ' sterillite' or title == 'Fennel seeds':
ax[x_plot][y_plot].set_ylim(0, 100)
ax[x_plot][y_plot].legend()
if x_axis_3 is not None and y_axis_3 is not None and label_3 is not None:
ax[x_plot][y_plot].plot(x_axis_3, y_axis_3, linewidth = 2, color = ([0.68, 0.74, 0.22]), label = label_3, marker = 'o')
ax[x_plot][y_plot].legend()
ax[x_plot][y_plot].set_xlim(xmin = 0.0)
ax[x_plot][y_plot].yaxis.set_tick_params(labelsize = 8)
ax[x_plot][y_plot].xaxis.set_tick_params(labelsize = 8)
ax[x_plot][y_plot].set_axisbelow(True)
ax[x_plot][y_plot].yaxis.grid(True)
ax[x_plot][y_plot].xaxis.grid(False)
ax[x_plot][y_plot].yaxis.set_major_formatter(FuncFormatter(format_y_tick_suffix))
ax[x_plot][y_plot].set_title(title, fontsize = 10, fontweight = "bold")
上面的代碼可以工作,但它太復雜了,無法理解。 有人可以建議我另一種寫作方式嗎? 我是python的新手,有人可以幫忙嗎。
只有兩個建議:
使用一個變量來保存ax[x_plot][y_plot]
,所以你不需要每次都重復。 例如::
a = ax[x_plot][y_plot] = plt.subplot2grid((3, 3), (x_plot, y_plot), colspan = column_span)
a.plot(...)
#etc.
而不是x_axis_1
等,創建一個結構來保存繪圖數據並將其作為列表提供:
from collections import namedtuple
PlotData = namedtuple('PlotData', 'x_axis y_axis label color')
# ...
plotdata_list = [Plotdata(x1, y1, label1, color21), ...]
get_plot(x_axis, y_axis, column_span, title, plotdata_list)
...並使用循環來繪制圖。
代碼(未經測試):
def get_plot(x_plot, y_plot, column_span, title, plotdata_list):
a = ax[x_plot][y_plot] = plt.subplot2grid((3, 3), (x_plot, y_plot), colspan = column_span)
for data in plotdata_list:
kwargs = {}
if data.label:
kwargs['label'] = data.label
a.plot(data.x_axis, data.y_axis, linewidth=2, color=data.color, marker='o', **kwargs)
if len(plotdata_list) > 1:
if title == ' sterillite' or title == 'Fennel seeds':
a.set_ylim(0, 100)
a.legend()
a.set_xlim(xmin = 0.0)
a.yaxis.set_tick_params(labelsize = 8)
a.xaxis.set_tick_params(labelsize = 8)
a.set_axisbelow(True)
a.yaxis.grid(True)
a.xaxis.grid(False)
a.yaxis.set_major_formatter(FuncFormatter(format_y_tick_suffix))
a.set_title(title, fontsize = 10, fontweight = "bold")
像這樣打電話(按照您的評論):
get_plot(0, 0, column_span=2, title='hp_accessories',
plotdata_list=[
# arguments: x, y, label, color
PlotData(hp_laser_ts, hp_laser, 'Lasers', [0.37, 0.23, 0.37]),
PlotData(hp_keyboard_ts, hp_keyboard, 'Keyboards', [0.44, 0.64, 0.69]),
PlotData(hp_mouse_ts, hp_mouse, 'Mice', [0.68, 0.64, 0.22]),
]
)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.