Python使用matplotlib从Counter堆积条形图

时间:2017-03-28 01:41:43

标签: python python-2.7 matplotlib counter bar-chart

我有一个大的csv文件。读取此文件,每预定义的行数返回Counter。作为一个例子:

counter = [Counter({(0, 1): 9, (1, 2): 8}), Counter({(1, 2): 99, (0, 1): 99}), Counter({(1, 2): 256, (0, 1): 189}), Counter({(1, 5): 473, (0, 1): 301})]

这是我使用的脚本。

import matplotlib.pyplot as plt
import numpy
from collections import Counter
counter = [Counter({(0, 1): 9, (1, 2): 8}), Counter({(1, 2): 99, (0, 1): 99}), Counter({(1, 2): 256, (0, 1): 189}), Counter({(1, 5): 473, (0, 1): 301})]

fig = plt.figure()
ax1 = fig.add_subplot(111)

N = len(counter)
ind = numpy.arange(N)

j = 0
while j in range(0, len(counter)):
    a, i = 0, 0
    frequencies = counter[j].values()
    names = counter[j].keys()
    while i in range(0, len(frequencies)):
        if i == 0:
            ax1.bar(ind, frequencies[i], label=names[i], width=0.25)
            a = frequencies[i]
        else:
            ax1.bar(ind, frequencies[i], label=names[i], width=0.25, bottom=a)
            a += frequencies[i]
        i += 1
    j += 1
labels = ["%s to %s" % (200, 202)]
ax1.set_xticks(numpy.arange(N))
ax1.set_xticklabels(labels)
ax1.set_ylabel("Frequency")
ax1.set_xlabel("Material Contact")
ax1.legend()

plt. show()

但是,它会返回错误消息:

  

ValueError:不兼容的大小:参数'height'必须是长度4或   标量

我认为这与ind数组相关。

为了解决这个问题,我在if语句中将ind更改为ind[j]。然而,最终的结果是很多酒吧都有很多颜色。正如预期的那样,颜色与各自的箱子无关。

ax1.bar(ind[j], frequencies[i], label=names[i], width=0.25)

获得的结果 Obtained Results

预期结果 Expected Results

更新: 一种可能的解决方案是从Counter构建一个数组。但是,这首先违反了反恐怖主义的概念。

1 个答案:

答案 0 :(得分:0)

所以唯一的答案是重新评估和重组数据

series = {}
for key in {key for keys in counter for key in keys}:
    series[key] = [(0 if key not in item else item[key]) for item in counter]

在[{3}}

上收到zivoni提供的帮助