python - 列表字典中 1's and 0' 的类别图

Question

我有一本看起来像这样的字典:

dict = {'a': [0,1,0,1,0,0,0,0,0], 'b': [1,1,1,1,1,1,1,1,0], 'c': [1,0,1,0,0,0,0,0,0] ...}

目标:制作一个类别条形图，其中 x 轴是键名，y 轴是 0 和 1 的比例。

key = dict.keys()
values = dict.values()

我知道使用 count() 可以得到 1 和 0 的数量

values.get('a').count(1)

如何使用列表字典绘制类别图？

Answer 1

您可以使用matplotlib :

import matplotlib.pyplot as plt


data = {
    'a': [0,1,0,1,0,0,0,0,0],
    'b': [1,1,1,1,1,1,1,1,0],
    'c': [1,0,1,0,0,0,0,0,0]
}

keys = list(data.keys()) # ["a", "b", "c"]

我将使用列表理解来构建一个包含计数的列表:

count_zero = [data[k].count(0) for k in keys] # [7, 1, 7]
count_ones = [data[k].count(1) for k in keys] # [2, 8, 2]

最后绘制数据:

# Create the bar plot
fig, ax = plt.subplots()
ind = list(range(1, len(data) + 1)) # [1, 2, 3, ...] horizontal location of the bars

bars = plt.bar(ind, count_zero)
bars = plt.bar(ind, count_ones, bottom=count_zero)

ax.set_xticks(ind)
ax.set_xticklabels(keys) # labels: a, b, c
ax.set_ylabel('Counts')
ax.set_title('StackOverflow')

plt.show()

结果: