绘图周围没有任何空格的 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 像素，您可以选择 100 dpi 和 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)