Python wordcloud周围没有任何空格

Question

我正在使用python3中的wordcloud模块并尝试保存一个只应该给我wordcloud图像的图形，而不会在云周围留下任何空白。我尝试了在stackexchange中提到的许多刻度，但它们没有用。下面是我的默认代码，它可以删除左侧和右侧的空白，但不会删除顶部和底部的空白。如果我将ax = plt.axes（[0,0,1,1]）中的其他两个值也设为0，那么我得到一个空图像。

wordcloud = WordCloud(font_path=None, width = 1500, height=500,  
            max_words=200,  stopwords=None, background_color='whitesmoke', max_font_size=None, font_step=1, mode='RGB', 
            collocations=True, colormap=None, normalize_plurals=True).generate(filteredText)

import matplotlib.pyplot as plt
fig = plt.figure()
ax = plt.axes([0,0,1,1])
plt.imshow(wordcloud, interpolation="nearest")
plt.axis('off')
plt.savefig('fig.png', figsize = (1500,500), dpi=300)

有人可以帮我解决这个问题吗？

Answer 1

wordcloud是一个图像，即一个像素阵列。 plt.imshow默认情况下使像素为正方形。这意味着除非图像具有与图形相同的纵横比，否则顶部和底部或左侧和右侧都会有空白区域。

您可以释放固定宽高比设置aspect="auto"，

plt.imshow(wc, interpolation="nearest", aspect="auto")

结果可能是不受欢迎的。

所以你真正想要的是使图形大小适应图像大小。 由于图像为1500 x 500像素，因此您可以选择dpi为100，图形尺寸为15 x 5英寸。

wc = wordcloud.WordCloud(font_path=None, width = 1500, height=500,  
            max_words=200,  stopwords=None, background_color='whitesmoke', max_font_size=None, font_step=1, mode='RGB', 
            collocations=True, colormap=None, normalize_plurals=True).generate(text)

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(15,5), dpi=100)
ax = plt.axes([0,0,1,1])
plt.imshow(wc, interpolation="nearest", aspect="equal")
plt.axis('off')
plt.savefig(__file__+'.png', figsize=(15,5), dpi=100)
plt.show()

<小时/> 最后使用matplotlib可能不是最好的选择。由于您只想保存图像，因此可以使用

from scipy.misc import imsave
imsave(__file__+'.png', wc)