Python：如何使用 Plotly 叠加直方图

Question

我在不同的列表中有两组数据。每个列表元素的值从 0:100 开始，并且元素重复。

例如：
first_data = [10,20,40,100,...,100,10,50]
second_data = [20,50,50,10,...,70,10,100]

我可以使用以下方法在直方图中绘制其中之一：

import plotly.graph_objects as go
.
.
.

fig = go.Figure()
fig.add_trace(go.Histogram(histfunc='count', x=first_data))
fig.show()

通过设置histfunc为'count'，我的直方图由从 0 到 100 的 x 轴和表示 中重复元素数量的条形组成first_data。

我的问题是：如何使用相同的“计数”直方图将第二组数据叠加在同一轴上？

Answer 1

实现此目的的一种方法是简单地添加另一条迹线，您就快到了！用于创建这些示例的数据集可以在本文的最后部分找到。

注意：
以下代码使用“较低级别”的plotly API，因为（个人）我觉得它更透明，并且使用户能够看到正在绘制的内容以及原因；graph_objects而不是依赖于和的便利模块express。

选项 1 - 重叠条形：

from plotly.offline import plot

layout = {}
traces = []

traces.append({'x': data1, 'name': 'D1', 'opacity': 1.0})
traces.append({'x': data2, 'name': 'D2', 'opacity': 0.5})

# For each trace, add elements which are common to both.
for t in traces:
    t.update({'type': 'histogram',
              'histfunc': 'count',
              'nbinsx': 50})

layout['barmode'] = 'overlay'

plot({'data': traces, 'layout': layout})

输出1：

选项 2 - 曲线图：

另一种选择是绘制分布曲线（高斯 KDE），如下所示。值得注意的是，该方法绘制的是概率密度，而不是计数。

X1, Y1 = calc_curve(data1)
X2, Y2 = calc_curve(data2)

traces = []
traces.append({'x': X1, 'y': Y1, 'name': 'D1'})
traces.append({'x': X2, 'y': Y2, 'name': 'D2'})

plot({'data': traces})

输出2：

关联calc_curve()函数：

from scipy.stats import gaussian_kde

def calc_curve(data):
    """Calculate probability density."""
    min_, max_ = data.min(), data.max()
    X = [min_ + i * ((max_ - min_) / 500) for i in range(501)]
    Y = gaussian_kde(data).evaluate(X)
    return(X, Y)

选项 3 - 绘制条形图和曲线：

或者，您始终可以使用 y 轴上的概率密度将这两种方法组合在一起。

layout = {}
traces = []

traces.append({'x': data1, 'name': 'D1', 'opacity': 1.0})
traces.append({'x': data2, 'name': 'D2', 'opacity': 0.5})

for t in traces:
    t.update({'type': 'histogram',
              'histnorm': 'probability density',
              'nbinsx': 50})

traces.append({'x': X1, 'y': Y1, 'name': 'D1'})
traces.append({'x': X2, 'y': Y2, 'name': 'D2'})

layout['barmode'] = 'overlay'

plot({'data': traces, 'layout': layout})  

输出3：

数据集：

以下是用于模拟 [0,100] 值数据集并创建这些示例的代码：

import numpy as np
from sklearn.preprocessing import MinMaxScaler

mms = MinMaxScaler((0, 100))
np.random.seed(4)
data1 = mms.fit_transform(np.random.randn(10000).reshape(-1, 1)).ravel()
data2 = mms.fit_transform(np.random.randn(10000).reshape(-1, 1)).ravel()