在python中创建垂直线的密度图答案

【问题标题】：Create a density plot of vertical lines in python在python中创建垂直线的密度图
【发布时间】：2019-08-15 15:48:43
【问题描述】：

我在一个大区域内有一堆包含坐标区间的数据，我想绘制这些数据，然后创建一个密度图，显示该区域中的区间线比其他区域多。

作为一个非常基本的示例，我刚刚为给定的间隔绘制了一些水平线。我真的找不到任何关于如何创建更好的区间图的好例子。我已经研究过seaborn，但我对此并不完全确定。所以在这里我刚刚创建了一个我正在尝试做的基本示例。

import numpy as np
import matplotlib.pyplot as plt

x1 = np.linspace(1, 30,100)
x2 = np.linspace(10,40,100)
x3 = np.linspace(2,50,100)
x4 = np.linspace(40,60,100)
x5 = np.linspace(30,78,100)
x6 = np.linspace(82,99,100)
x7 = np.linspace(66,85,100)
x = [x1,x2,x3,x4,x5,x6,x7]
y = np.linspace(1,len(x),len(x))
fig, ax = plt.subplots()
for i in range(len(x)):
    ax.hlines(y[i], xmin=x[i][0], xmax=x[i][-1], linewidth=1)
plt.xlim(-5,105)
plt.show()

然后我想创建一个间隔重叠数的密度图。任何人都可以对如何进行此操作有任何建议吗？

感谢您的帮助和建议

【问题讨论】：

标签： python matplotlib density-plot horizontal-line

【解决方案1】：

这似乎是你想要的：

def count(xi):
    samples = np.linspace(0, 100, 101)
    return (xi[0] < samples) & (samples <= xi[-1])

is_in_range = np.apply_along_axis(count, arr=x, axis=1)
density = np.sum(is_in_range, axis=0)

一般的想法是输出一些linspace，然后检查这些坐标是否在数组x 的范围内——这就是函数count 所做的。然后apply_along_axis 在您的数组x 中的每一行（即每个一维数组）上运行此函数。

这是我在绘制density 时得到的结果：

您可能需要调整count 函数中的<= 和< 符号以根据需要处理边缘。

如果您的实际数据有不同的格式，或者如果一个数组中有多个区间，您将需要对此进行调整。

【讨论】：