2016-09-14 47 views
1

我試圖用Matplotib創建多個單獨的圖塊,然後將這些圖塊保存到一個PDF文檔中。這裏是我的代碼:使用Matplotlib創建多個圖塊

pdf = matplotlib.backends.backend_pdf.PdfPages('Activity_Report.pdf') 

fig1 = plt.figure(1) 
fig1.figure(figsize=(11.69, 8.27)) 
ax1 = fig1.add_subplot(111) 


# ******** product 1 ******** 
ax1.plot(Prod_01['Date'], Prod_01['Orders'], marker='o', label='Orders', color='navy', linewidth='2') 
ax1.plot(Prod_01['Date'], Prod_01['Orders_MA'], linestyle='--', label='Orders (10-d)', color='darkblue', linewidth='2') 

ax1.plot(Prod_01['Date'], Prod_01['Volume'], marker='o', label='Volume', color='firebrick', linewidth='2') 
ax1.plot(Prod_01['Date'], Prod_01['Volume_MA'], linestyle='--', label='Volume (10-d)', color='firebrick', linewidth='2') 

ax1.plot(Prod_01['Date'], Prod_01['Pass'], marker='o', label='Pass', color='darkgreen', linewidth='2') 
ax1.plot(Prod_01['Date'], Prod_01['Pass_MA'], linestyle='--', label='Pass (10-d)', color='darkgreen', linewidth='2') 

ax1.plot(Prod_01['Date'], Prod_01['Request'], marker='o', label='Request', color='cyan', linewidth='2') 
ax1.plot(Prod_01['Date'], Prod_01['Request_MA'], linestyle='--', label='Request (10-d)', color='cyan', linewidth='2') 

ax1.set_title('Prod_01', fontsize='20', rasterized=True) 
ax1.tick_params(axis='both', which='major', labelsize='10') 
ax1.legend(loc='upper left', fontsize='10') 

ax1.get_yaxis().set_major_formatter(tkr.FuncFormatter(lambda x, p: format(int(x), ','))) 
ax1.xaxis.set_major_formatter(custom_x_axis_format) 

# ******** product 2 ******** 

fig2 = plt.figure(2) 
fig2.figure(figsize=(11.69, 8.27)) 
ax2 = fig2.add_subplot(111) 

ax2.plot(Prod_02['Date'], Prod_02['Order'], marker='o', label='Order', color='navy', linewidth='2') 
ax2.plot(Prod_02['Date'], Prod_02['Order_MA'], linestyle='--', label='Order (10-d)', color='darkblue', linewidth='2') 

ax2.plot(Prod_02['Date'], Prod_02['Volume'], marker='o', label='Volume', color='firebrick', linewidth='2') 
ax2.plot(Prod_02['Date'], Prod_02['Volume_MA'], linestyle='--', label='Volume (10-d)', color='firebrick', linewidth='2') 

ax2.plot(Prod_02['Date'], Prod_02['Pass'], marker='o', label='Pass', color='darkgreen', linewidth='2') 
ax2.plot(Prod_02['Date'], Prod_02['Pass_MA'], linestyle='--', label='Pass (10-d)', color='darkgreen', linewidth='2') 

ax2.plot(Prod_02['Date'], Prod_02['Request'], marker='o', label='Request', color='cyan', linewidth='2') 
ax2.plot(Prod_02['Date'], Prod_02['Request_MA'], linestyle='--', label='Request (10-d)', color='cyan', linewidth='2') 

ax2.set_title('Prod_02', fontsize='20', rasterized=True) 
ax2.tick_params(axis='both', which='major', labelsize='10') 
ax2.legend(loc='upper left', fontsize='10') 

ax2.get_yaxis().set_major_formatter(tkr.FuncFormatter(lambda x, p: format(int(x), ','))) 
ax2.xaxis.set_major_formatter(custom_x_axis_format) 


pdf.savefig() 
pdf.close() 

我遇到的問題是,這個代碼打印到人物(而不是兩個獨立人物也就是說,product 1product 2在一個單一的數字繪製。 (即在彼此的頂部)。我試圖用Figure()大獨立地塊(在StackOverflow上幾個帖子可以看到),但似乎並沒有工作。

這可能是因爲我有fig1ax1fig2ax2已定義inco直接(對Python來說是新手,我仍然不明白他們的使用是100%)。

有沒有人看到爲什麼這段代碼生成一個單一的情節,而不是有意的兩個獨立的情節?

在此先感謝!

+0

那也許是因爲你只調用pdf.savefig()一次? – benten

+0

您需要將這些數據專門保存到PDF中,例如'pdf.savefig(fig1)'和'pdf.savefig(fig2)' –

+0

謝謝您(@benten和@Paul H)糾正我的錯誤。代碼現在按預期工作! – equanimity

回答

3

你的代碼有點過於複雜。這兩條線是多餘的:

fig1 = plt.figure(1) 
fig1.figure(figsize=(11.69, 8.27)) 

我只是:

FIGSIZE = (11.69, 8.27) 
fig1, ax1 = plt.subplots(figsize=FIGSIZE) 
# plot things 

fig2, ax2 = plt.subplots(figsize=FIGSIZE) 
# plot more things 

# etc etc 

pdf = matplotlib.backends.backend_pdf.PdfPages('Activity_Report.pdf') 
for fig in [fig1, fig2, ...]: 
    pdf.savefig(fig) 

plt.close('all')