mpld3插件设置x轴刻度

3
我正在尝试制作一个mpld3插件,以便当我在散点图的圆圈上鼠标悬停时,它会显示正弦线,但x轴从线开始处开始,并以线结束处结束。要达到这个目的,应该调用哪个方法?
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
import mpld3
from mpld3 import plugins, utils

mpld3.enable_notebook()

class LinkedView(plugins.PluginBase):
    """A simple plugin showing how multiple axes can be linked"""

    JAVASCRIPT = """
    mpld3.register_plugin("linkedview", LinkedViewPlugin);
    LinkedViewPlugin.prototype = Object.create(mpld3.Plugin.prototype);
    LinkedViewPlugin.prototype.constructor = LinkedViewPlugin;
    LinkedViewPlugin.prototype.requiredProps = ["idpts", "idline", "data"];
    LinkedViewPlugin.prototype.defaultProps = {}
    function LinkedViewPlugin(fig, props){
        mpld3.Plugin.call(this, fig, props);
    };

    LinkedViewPlugin.prototype.draw = function(){
      var pts = mpld3.get_element(this.props.idpts);
      var line = mpld3.get_element(this.props.idline);
      var data = this.props.data;

      pts.elements().on("mouseover", function(d, i){

        var begin = d[0]-2;
        var end = d[0]+2;
        console.log("Show from " + begin + " to " + end);
        var truncated = Array();
        for (var row = 0; row < data[i].length ; ++row){

            var x = data[i][row][0];
            if ( begin <= x && x <= end){
                truncated.push(data[i][row]);
            }
        }

        line.data = truncated;
        line.elements().transition()
            .attr("d", line.datafunc(line.data))
            .style("stroke", this.style.fill);

      });
    };
    """

    def __init__(self, points, line, linedata):
        if isinstance(points, matplotlib.lines.Line2D):
            suffix = "pts"
        else:
            suffix = None

        self.dict_ = {"type": "linkedview",
                      "idpts": utils.get_id(points, suffix),
                      "idline": utils.get_id(line),
                      "data": linedata}

fig, ax = plt.subplots(2)

# scatter periods and amplitudes
np.random.seed(0)
P =np.random.random(size=20)*10
A = np.random.random(size=20)
x = np.linspace(0, 10, 100)
data = np.array([[x, Ai * np.sin(x / (Pi/10))]

for (Ai, Pi) in zip(A, P)])
points = ax[1].scatter(P, A, c=(P/10 + A),s=200, alpha=0.5)
ax[1].set_xlabel('Period')
ax[1].set_ylabel('Amplitude')

# create the line object
lines = ax[0].plot(x, 0 * x, '-w', lw=3, alpha=0.5)
ax[0].set_ylim(-1, 1)

ax[0].set_title("Hover over points to see lines")

# transpose line data and add plugin
linedata = data.transpose(0, 2, 1).tolist()
plugins.connect(fig, LinkedView(points, lines[0], linedata))

mpld3.display()
1个回答

1
mpld3.js中有一个名为set_axlim的javascript函数。在您的javascript代码中,您需要使用您想要设置的限制调用pts.ax.set_axlim此处是调用该方法的示例。

谢谢,有了你的帮助,这真的很容易让它工作了。 "line.ax.set_axlim([begin,end],[-1,1]);" - Ignacio Tartavull

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接