使用元组元素从列表中生成频率直方图

时间:2017-07-13 12:30:41

标签: python matplotlib plot distribution frequency

我想制作一个单词频率分布,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()

2 个答案:

答案 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')

enter image description here

答案 1 :(得分:4)

您无法直接将word传递给matplotlib.pyplot.bar。但是,您可以为bar创建索引数组,然后使用matplotlib.pyplot.xtickswords替换为这些索引:

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()

enter image description here

创建forword的{​​{1}} - 循环也可以替换为简单的frequency和列表解包:

zip