如何根据python中的频率和值为每个用户id创建词云图

Question

我正在尝试根据单词和计数频率为每个用户创建每个单词云图，我想将单词云图像路径的输出与 UID.Do 一起存储在数据框中 UID.Do 我需要应用分组依据吗？任何帮助将不胜感激。

|UID         |word                     |count          
=================================================
|ccf878ec9315|RT                       |28               
|ccf878ec9315|Newpin                   |6                
|ccf878ec9315|Benefit Bond             |6                
|ccf878ec9315|Covid                    |5                
|ccf878ec9315|Blues                    |5                
|ccf878ec9316|TPG                      |10               
|ccf878ec9316|Learn                    |8                
|ccf878ec9316|An                       |6                
|ccf878ec9317|GIINs Market Roadmap     |9                
|ccf878ec9317|amp                      |5                
|ccf878ec9317|Varsity                  |3                
|ccf878ec9318|International Womens Day |10               
|ccf878ec9318|Solving                  |8                
|ccf878ec9318|Hadewych                 |4                
|ccf878ec9319|GIF16                    |4               
|ccf878ec9319|Kuyper                   |9                
|ccf878ec9320|Impact Investments       |8                
|ccf878ec9320|Climate                  |3

我尝试使用频率计数，它包含整个数据集的所有数据。但是我无法为每个UID创建词云图。

from wordcloud import WordCloud

wc = WordCloud(width=800, height=400, max_words=200).generate_from_frequencies(data)
import matplotlib.pyplot as plt

plt.figure(figsize=(10, 10))
plt.imshow(wc, interpolation='bilinear')
plt.axis('off')
plt.show()

Answer 1

generate_from_frequencies(data) 中的 data 需要是类似于 {'RT': 28, 'Newpin': 6, 'Benefit Bond': 6, 'Covid': 5, ...} 的 Python 字典。这是从给定数据帧创建这样一个字典的方法：

import pandas as pd
import numpy as np

df = pd.DataFrame({'UID': ['ccf878ec9315', 'ccf878ec9315', 'ccf878ec9315', 'ccf878ec9315', 'ccf878ec9315',
                           'ccf878ec9316', 'ccf878ec9316', 'ccf878ec9316', 'ccf878ec9317', 'ccf878ec9317',
                           'ccf878ec9317', 'ccf878ec9318', 'ccf878ec9318', 'ccf878ec9318', 'ccf878ec9319',
                           'ccf878ec9319', 'ccf878ec9320', 'ccf878ec9320'],
                   'word': ['RT', 'Newpin', 'Benefit Bond', 'Covid', 'Blues', 'TPG', 'Learn', 'An',
                            'GIINs Market Roadmap', 'amp', 'Varsity', 'International Womens Day', 'Solving', 'Hadewych',
                            'GIF16', 'Kuyper', 'Impact Investments', 'Climate'],
                   'count': [28, 6, 6, 5, 5, 10, 8, 6, 9, 5, 3, 10, 8, 4, 4, 9, 8, 3]})
data = {wrd: cnt for wrd, cnt in zip(df['word'], df['count'])}

from wordcloud import WordCloud

wc = WordCloud(width=800, height=400, max_words=200).generate_from_frequencies(data)
import matplotlib.pyplot as plt

plt.figure(figsize=(10, 10))
plt.imshow(wc, interpolation='bilinear')
plt.axis('off')
plt.show()

或者，完全在 pandas 内创建字典：

data = df.set_index('word')['count'].to_dict()

为每个 UID 创建一个词云：

uids = np.unique(df['UID'])

fig, axes = plt.subplots(nrows=(len(uids)+2)//3, ncols=3, figsize=(20, 8),
                         gridspec_kw={'hspace': 0.05, 'wspace': 0.05, 'left': 0.01, 'right': 0.99, 'top': 0.99, 'bottom': 0.01})for uid, ax in zip(uids, axes.ravel()):
    data = df[df['UID'] == uid].set_index('word')['count'].to_dict()

    wc = WordCloud(width=800, height=400, max_words=200).generate_from_frequencies(data)
    ax.imshow(wc, interpolation='bilinear')
    ax.set_title(f'UID = {uid}')
    ax.axis('off')
plt.show()

如何根据python中的频率和值为每个用户id创建词云图

How to create wordcloud chart for each userid based on frequency and value in python

python

python-3.x

matplotlib

word-cloud