为数字列生成 WordCloud

Question

我想知道是否可以在数字数据框列上生成词云，其值可用于在词云中进行解释。

我试过这样做：

text = df['tag'].value_counts().to_dict()

wordcloud = WordCloud().generate_from_frequencies(text)

但出现此错误：

TypeError: 'int' 类型的参数不可迭代

有人可以帮忙吗？ 谢谢

Answer 1

将您的 tag 列转换为字符串：.astype(str)

import pandas as pd
from wordcloud import WordCloud
import matplotlib.pyplot as plt

df = pd.DataFrame({'tag': [1, 1, 2, 3, 3, 3]})

text = df['tag'].astype(str).value_counts().to_dict()
wc = WordCloud().generate_from_frequencies(text)

plt.imshow(wc)
plt.show()