将 3 元组列表转换为 matplot 堆积条形图的列表列表

Convert a list of 3-Tuples in to list of list for a matplot stacked bar chart

我有一个包含 3 个元组的列表,用于装箱问题的解决方案,如下所示:

sorted_sol = [(0, 1, 170), (1, 1, 250), (2, 1, 250), (3, 1, 62), (3, 2, 30), (4, 1, 62), (4, 3, 62), (5, 2, 122), (6, 1, 212)]

例如,第一个三元组表示从长度 0 切割 1 @ 170

我正在尝试将 3 元组列表转换为 matplot 堆叠条形图的列表列表,但我正在努力处理循环逻辑。结果应该是。

import matplotlib.pyplot as plt
import numpy as np
bars = list(set([int(i[0]) for i in sorted_sol]))
#loop logic here to end up with data
b1 = [170, 250, 250, 62, 62, 122, 212]
b2 = [0,   0,   0,   30, 62, 122, 0]
b3 = [0,   0,   0,   30, 62, 0,   0]
b4 = [0,   0,   0,   0,  62, 0,   0]
data =[b1, b2, b3, b4]
for c in range(0, 3):
    if c == 0:
        plt.bar(bars, data[c])
    else:
        plt.bar(bars, data[c], bottom=data[c-1])
plt.show()

此外,底部 属性 似乎对我不起作用,因为它似乎没有正确堆叠条形图 3 和 4。

我没有遵循你从 sorted_sol 到列表 b1,b2,b3,b4 的逻辑。这对我来说似乎很奇怪,因为你的 sorted_sol 只有 3 个值 62data 有 5 个这样的值。

无论如何,一旦达到 data,您可以考虑使用 pandas 来实现堆叠栏功能:

import pandas as pd
df = pd.DataFrame(data).T
df.plot.bar(stacked=True)

输出:

  • sort 元组的(索引一,索引零)列表
  • group 元组之一的索引列表
  • 每个组都是一个 b 列表
    • 元组索引 0 是(新)b 列表中的索引

我能看到的最好的,但似乎不适合元组列表和 b 列表

IIUC,这样的东西应该行得通

dx = max(x[0] for x in sorted_sol) + 1
predata = []
for x in range(dx):
    col_data = [tup[1:] for tup in sorted_sol if tup[0] == x]
    temp = [n * [y] for n, y in col_data]
    predata.append([i for sublist in temp for i in sublist])

dy = max(len(x) for x in predata)
data = [[i.pop() if i else 0 for i in predata] for _ in range(dy)]

输出:

[[170, 250, 250, 30, 62, 122, 212],
 [0, 0, 0, 30, 62, 122, 0],
 [0, 0, 0, 62, 62, 0, 0],
 [0, 0, 0, 0, 62, 0, 0]]

您的绘图不起作用的原因是因为 bottom 需要是数据的累积总和。尝试类似的东西:

bottom = 7 * [0]
for i in range(4):
    plt.bar(range(7), data[c], bottom=bottom)
    bottom = [sum([a, b]) for a, b in zip(data[c], bottom)]

如果你使用numpy,最后一行的加法可以简单地是data[i] + bottom