我想制作一个单词频率分布,x轴上的字和y轴上的频率计数。
我有以下列表:
example_list = [('dhr', 17838), ('mw', 13675), ('wel', 5499), ('goed', 5080),
('contact', 4506), ('medicatie', 3797), ('uur', 3792),
('gaan', 3473), ('kwam', 3463), ('kamer', 3447),
('mee', 3278), ('gesprek', 2978)]
我尝试首先将其转换为pandas DataFrame,然后使用pd.hist()
,如下例所示,但我无法弄明白并认为它实际上是直接的但可能我缺少东西。
import numpy as np
import matplotlib.pyplot as plt
word = []
frequency = []
for i in range(len(example_list)):
word.append(example_list[i][0])
frequency.append(example_list[i][1])
plt.bar(word, frequency, color='r')
plt.show()
答案 0 :(得分:5)
使用pandas:
import pandas as pd
import matplotlib.pyplot as plt
example_list = [('dhr', 17838), ('mw', 13675), ('wel', 5499), ('goed', 5080), ('contact', 4506), ('medicatie', 3797), ('uur', 3792), ('gaan', 3473), ('kwam', 3463), ('kamer', 3447), ('mee', 3278), ('gesprek', 2978)]
df = pd.DataFrame(example_list, columns=['word', 'frequency'])
df.plot(kind='bar', x='word')
答案 1 :(得分:4)
您无法直接将word
传递给matplotlib.pyplot.bar
。但是,您可以为bar
创建索引数组,然后使用matplotlib.pyplot.xticks
将words
替换为这些索引:
import numpy as np
import matplotlib.pyplot as plt
indices = np.arange(len(example_list))
plt.bar(indices, frequency, color='r')
plt.xticks(indices, word, rotation='vertical')
plt.tight_layout()
plt.show()
创建for
和word
的{{1}} - 循环也可以替换为简单的frequency
和列表解包:
zip