如何生成带有截断条形的条形图？

Question

假设我有两个列表。 我从原始 2dim 列表中提取了最小、最大列表。

循环原始列表，我提取列表。提取列表的值为 -100s 和列表的最小值

例如，从 [0, 0, 1, 3, 10] 列表，返回 [-100, -100, 0]。

在这里，-100 的数量表示列表中最小值的数量（在本例中为 0 的数量）。 0 是列表的最小值。

original = [ [0, 0, 1, 3, 10], [4, 2, 2, 5, 10, 15], [3, 0, 0, 0, 0, 4, 10] ...]
min = [-100, -100, 0, -100, -100, 2, -100, -100, -100, -100 0, ...]
max = [-100, -100, 10, -100, -100, 15, -100, -100, -100, -100, 10, ...]

有了这个列表，我想绘制如下条形图。

如果遇到-100，则通过那个时间。如果不是，则从最小值到最大值绘制条形图。

我还想根据标准给出不同的颜色。

我在 matplotlib example page 或谷歌的搜索结果中找不到很好的例子。

我可以得到这方面的提示吗？

Answer 1

以下代码使用matplotlib's colourmap 实现了预期的结果：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors

min_vals = [-100, -100, 0, -100, -100, 2, -100, -100, -100, -100, 0]
max_vals = [-100, -100, 10, -100, -100, 15, -100, -100, -100, -100, 10]
colours = [1, 1, 2, 1, 1, 3, 1, 1, 1, 1, 2]

w = len(min_vals)
h = max(max_vals) - min(x for x in min_vals if x != -100)

arr = np.ones((h, w), dtype="int")

for i, vals in enumerate(zip(min_vals, max_vals)):
    if -100 not in vals:
        arr[vals[0]:vals[1]:,i] = np.array([colours[i]]*(vals[1]-vals[0]))

cmap = colors.ListedColormap(['white', 'green', 'red'])
fig, ax = plt.subplots(figsize=(6,6))

ax.imshow(arr, cmap=cmap)
plt.gca().invert_yaxis()
plt.yticks(range(h))
plt.xticks(range(w))
plt.show()

输出：