前言

本篇博客主要解決在使用pandas繪制圖像并保存時，由于標(biāo)簽太長，導(dǎo)致坐標(biāo)軸上的標(biāo)簽顯示不全的問題。剛遇到問題時調(diào)整了一下圖片大小，然鵝并沒有卵用，于是乎就檢索了一下問題，發(fā)現(xiàn)沒有解決pandas的。查詢無果后，查看了一下官方文檔，頓悟，這不就是matplotlib嘛，換了一個關(guān)鍵字再查詢，果然，就是。所以本篇同樣適用于解決matplotlib繪制圖像時出現(xiàn)的這個問題。

	# 部分代碼
	df_sparsity = distiller.weights_sparsity_summary(resnet20)
	df_sparsity_tmp = df_sparsity[['NNZ (dense)', 'NNZ (sparse)']]
	ax = df_sparsity_tmp.iloc[0:-1].plot(kind='bar', figsize=[30,10], title="Sparse vs. Dense\n(element-wise)")
	ax.set_xticklabels(df_sparsity.Name, rotation=90)

	fig = ax.get_figure()
	fig.savefig('output.png')

1. 問題描述

使用pandas的DataFrame存儲數(shù)據(jù)并繪制圖像，橫坐標(biāo)軸的標(biāo)簽是ResNet20的各個layer的名字，像module.layer1.0.conv1.weight這樣，由于太長，導(dǎo)致繪制出來的圖像是這樣的：

坐標(biāo)軸的標(biāo)簽沒有顯示全，于是就嘗試改變圖像的大小：

	# 改變下高度
	ax = df_sparsity_tmp.iloc[0:-1].plot(kind='bar', figsize=[30,50], title="Sparse vs. Dense\n(element-wise)")

于是圖像就變成了下面這樣，雖然有效果，但圖片被拉長，依舊不美觀：

2. 問題原因

既然調(diào)整繪制圖像尺寸之后繪制出來的圖片依舊不符合預(yù)期，那就只能看看官方的文檔了----pandas.DataFrame.plot()，看看是不是還有其他參數(shù)，能夠使圖像的大小及位置不變，坐標(biāo)軸的繪制空間適當(dāng)調(diào)整。然后就發(fā)現(xiàn)了這個：

函數(shù)返回了一個matplotlib.axes.Axes，既然是matlitplot，那問題就很容易解決了，matplotlib老搞這樣的事。查看一下matplotlib的文檔----matplotlib.pyplot.savefig()，可以看到：

默認(rèn)情況下，僅保存圖形給定的部分，如果設(shè)置為tight，將嘗試保存更緊致的圖形。

4. 解決方法

在保存圖形是加入bbox_inches參數(shù)：

	# 設(shè)置tight bbox
	fig.savefig('output.png', bbox_inches='tight')

再次保存圖形，此時的就很ok了：

結(jié)束語