python - 是否有 pandas 自相关图方法的 Bokeh 版本？

Question

假设我们有一个名为"series" 的时间序列对象。我知道使用 autocorrelation_plot() 方法绘制 series 对象的滞后和自相关维度非常容易。

代码如下:

from matplotlib import pyplot
from pandas.tools.plotting import autocorrelation_plot
autocorrelation_plot(series)
pyplot.show()

这是 Pandas 图:

有没有办法使用 Bokeh 服务器获得相同的情节？

Answer 1

是的，有。我编写的代码可以为您提供与 pandas autocorrelation_plot() 方法相同的结果。

代码如下:

from bokeh.layouts import column
from bokeh.plotting import figure, curdoc
import timeseries_model_creator  # to get data
import numpy as np

TimeSeriesModelCreator = timeseries_model_creator.TimeSeriesModelCreator()
series = TimeSeriesModelCreator.read_csv()  # time series object

def get_autocorrelation_plot_params(series):
    n = len(series)
    data = np.asarray(series)

    mean = np.mean(data)
    c0 = np.sum((data - mean) ** 2) / float(n)

    def r(h):
        return ((data[:n - h] - mean) *
                (data[h:] - mean)).sum() / float(n) / c0
    x = np.arange(n) + 1
    y = map(r, x)
    print "x : ", x, " y : ", y
    z95 = 1.959963984540054
    z99 = 2.5758293035489004
    return n, x, y, z95, z99

n, x, y, z95, z99 = get_autocorrelation_plot_params(series)

auto_correlation_plot2 = figure(title='Time Series Auto-Correlation', plot_width=1000,
                                plot_height=500, x_axis_label="Lag", y_axis_label="Autocorrelation")

auto_correlation_plot2.line(x, y=z99 / np.sqrt(n), line_dash='dashed', line_color='grey')
auto_correlation_plot2.line(x, y=z95 / np.sqrt(n), line_color='grey')
auto_correlation_plot2.line(x, y=0.0, line_color='black')
auto_correlation_plot2.line(x, y=-z95 / np.sqrt(n), line_color='grey')
auto_correlation_plot2.line(x, y=-z99 / np.sqrt(n), line_dash='dashed', line_color='grey')

auto_correlation_plot2.line(x, y, line_width=2)
auto_correlation_plot2.circle(x, y, fill_color="white", size=8)  # optional

curdoc().add_root(column(auto_correlation_plot2))

这是 Bokeh 图: