python - 如何使用 Bokeh 绘制阶跃函数？

Question

在 matplotlib 中创建一个 step 函数，你可以这样写：

import matplotlib.pyplot as plt

x = [1,2,3,4] 
y = [0.002871972681775004, 0.00514787917410944, 
     0.00863476098280219, 0.012003316194034325]

plt.step(x, y)
plt.show()

如何使用 Bokeh 制作类似的图表？

Answer 1

Bokeh 有一个Step内置的字形版本0.12.11：

from bokeh.plotting import figure, output_file, show

output_file("line.html")

p = figure(plot_width=400, plot_height=400)

# add a steps renderer
p.step([1, 2, 3, 4, 5], [6, 7, 2, 4, 5], line_width=2, mode="center")

show(p)

Answer 2

您可以line对阵列进行一些手动操作。请记住轴列表的交错。y此外，我们希望（我猜）稍微修改 x 值以完成左右限制。让我提供以下内容：

from bokeh.plotting import figure
import numpy as np
f = figure()
#
# Example vectors (given):
x = [1,2,3,4]
y = [0.002871972681775004, 0.00514787917410944, 
     0.00863476098280219, 0.012003316194034325]
#
_midVals = np.diff(x)/2.0
xs = set(x[:-1]-_midVals).union(x[1:]+_midVals)
xl = list(xs)
xl.sort()
xx = xl[1:]+xl[:-1]
xx.sort()
yy = y+y
yy[::2] = y
yy[1::2] = y
#
# assert/print coordinates
assert len(xx)==len(yy)
for _p in zip(xx,yy):
    print _p
#
# Plot!
f.line(x=xx, y=yy)
show(f)

我们的“健全性检查”打印的出口应该是：

(0.5, 0.002871972681775004)
(1.5, 0.002871972681775004)
(1.5, 0.00514787917410944)
(2.5, 0.00514787917410944)
(2.5, 0.00863476098280219)
(3.5, 0.00863476098280219)
(3.5, 0.012003316194034325)
(4.5, 0.012003316194034325)

和情节：

希望对到达这里的人有所帮助。

Answer 3

我写了一个小函数，它使用 line 来完成这个，matplotlib 风格比@Brandt 的答案多一点。在 jupyter 笔记本中，这是

def step(fig, xData, yData, color=None, legend=None):
    xx = np.sort(list(xData) + list(xData))
    xx = xx[:-1]
    yy = list(yData) + list(yData)
    yy[::2] = yData
    yy[1::2] = yData
    yy = yy[1:]
    fig.line(xx, yy, color=color, legend=legend)
    return fig
# example
import bokeh.plotting as bok
import bokeh
bokeh.io.output_notebook()

x = [1,2,3,4]
y = [0.002871972681775004, 0.00514787917410944, 
     0.00863476098280219, 0.012003316194034325]
f = bok.figure()
f = step(f, x, y)
bok.show(f)

给你

当然，您可以将 output_notebook() 替换为您希望的任何输出格式。