python - 如何在 Python 中绘制具有可变宽度 bin 的直方图？

Question

假设我有数据[1,2,3, 7,8,9,9, 20,30,40,100,1000]，我想用 Python 为其绘制直方图。我关心的所有 bin 是 [0,5]、[5,10] 和 [10, +∞)。我该怎么做？

当然，以下不会这样做。

import matplotlib.pyplot as plt

data = [1,2,3, 7,8,9,9, 20,30,40,100,1000]
plt.figure()
plt.hist(data, bins=5, color="rebeccapurple")
plt.show()

Answer 1

如果强制显示具有自定义 x 范围的直方图，您可能需要先处理您的数据。

我制作了一个范围列表和 x_ticklabels 以显示带有范围的 x 轴。

import matplotlib.pyplot as plt
import numpy as np

data = [1,2,3, 7,8,9,9, 20,30,40,100,1000,500,200]
data = np.array(data)
bin_range = [
    [0, 5],
    [5, 10],
    [10, 10000] # enough number to cover range
]

data2plot = np.zeros(len(bin_range))

for idx, (low, high) in enumerate(bin_range):
    data2plot[idx] = ((low <= data) & (data < high)).sum()
    
fig = plt.figure()
ax = fig.add_subplot(111)
ax.bar(range(len(bin_range)), data2plot)

x_labels = [
    f"{low}~{high}" for idx, (low, high) in enumerate(bin_range)
]

ax.set_xticks(range(len(bin_range)))
ax.set_xticklabels(x_labels)
plt.show()