如何在Python中创建一个堆叠的条形图，其中每个变量的值之间存在间隙

PPM 发表于 Dev

PPM

我在Python中创建堆积的条形图时遇到问题。我有3个变量的数据，如下所示：

A=[3,5,7]

B=[4,5,7]

C=[2,3,4,5,6,7]

我想创建一个条形图，其中每个变量的值都如下所示

条形图，每个变量的值之间存在间隙：

有人可以帮我这个图表吗？非常感谢你。

亚历山大·B

实际上，您想绘制一个颜色表或接近甘特图的图形。我不知道在Python中真正方便的方法。

一种解决方案是使用matplotlib.pyplot.grid （doc）。该讨论提供了一种解决方案。

另一个解决方案是研究plotly软件包。它为表格和甘特图（doc）提供了非常漂亮的输出。

在这里，我将向您展示hbar来自matplotlib的类似输出。主要思想是重建单元格网格。每一行代表一类（例如A，B或C）。每行由相同数量的单元格组成。每个单元具有相同的宽度。单元格的颜色由数据的二进制翻译定义。

要调整x-axis标签，只需手动移动它们即可。

# Import module
import seaborn
import matplotlib.pyplot as plt
import numpy as np

##########################
#       Your input       #
##########################

A = [3, 5, 7]
B = [4, 5, 7]
C = [2, 3, 4, 5, 6, 7]


##########################
# Prepra the data        #
##########################
my_list = [A, B, C]
segments = 8
cell_width = 1000
nb_x_range = [i * cell_width for i in range(2, segments)]
classes = ["A", "B", "C"]
colors_def = {'A': {0: "w", 1: "b"},
              'B': {0: "w", 1: "g"},
              'C': {0: "w", 1: "y"},
              }

def create_data(my_list):
    data = np.zeros((segments + 1, len(classes)))
    for i, sub_list in enumerate(my_list):
        for elt in sub_list:
            data[elt, i] = 1
    return data

data = create_data(my_list)
print(data)
# [[0. 0. 0.]
#  [0. 0. 0.]
#  [0. 0. 1.]
#  [1. 0. 1.]
#  [0. 1. 1.]
#  [1. 1. 1.]
#  [0. 0. 1.]
#  [1. 1. 1.]
#  [0. 0. 0.]]

y_pos = np.arange(len(classes))
# left alignment of data starts at zero
left = np.zeros(len(my_list)) - cell_width/2


##########################
#       Create plot      #
##########################
# Set sea born for prettier graph
seaborn.set()
# create figure
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111)
# Set X axis (shifted)
ax.set_xlim((min(nb_x_range)-cell_width/2, max(nb_x_range)+cell_width/2))

# For each cell
for d in data:
    # Define color for each row
    colors = [colors_def[classes[i]][bi] for i, bi in enumerate(d)]
    # e.g. colors = [colors_def["A"][d[0]], colors_def["B"][d[1]], colors_def["C"][d[2]]]

    # Draw cell
    ax.barh(y_pos, cell_width,
            height=1,           # Heights of horizontal bars
            color=colors,       # Colors
            left=left)          # Left padd from y-axis
    # Update margin
    left += cell_width

# Axis graduation
ax.set_yticks(y_pos)
ax.set_xticks(nb_x_range)
ax.set_yticklabels(classes)
ax.set_xlabel('Stacked bar')

plt.show()

输出如下：