Chaco - 获取多个数据系列以使用相同的轴和地图

时间:2014-05-24 09:43:39

标签: enthought traitsui chaco

我试图在一个地块上绘制几个数据集。

每个数据集可以表示为x系列(索引)和几个y系列(值)。 x和y数据序列的范围在每个数据集中可以不同。我想在一个图上显示几个这些数据集。但是,当我简单地将第二个绘图对象添加到第一个绘图对象时(见下文),它为它绘制了第二个轴,它嵌套在绘图中。

我希望两个图共享相同的轴,并且轴边界要更新以适合所有数据。实现这一目标的最佳方法是什么?我正在努力在文档中找到相关主题。

感谢您的帮助。下面的代码突出了我的问题。

# Major library imports
from numpy import linspace
from scipy.special import jn

from chaco.example_support import COLOR_PALETTE
# Enthought library imports
from enable.api import Component, ComponentEditor
from traits.api import HasTraits, Instance
from traitsui.api import Item, Group, View

# Chaco imports
from chaco.api import ArrayPlotData, Plot
from chaco.tools.api import BroadcasterTool, PanTool, ZoomTool
from chaco.api import create_line_plot, add_default_axes

def _create_plot_component():    
    # Create some x-y data series to plot
    x = linspace(-2.0, 10.0, 100)
    x2 =linspace(-5.0, 10.0, 100)

    pd = ArrayPlotData(index = x)
    for i in range(5):
        pd.set_data("y" + str(i), jn(i,x))

    #slightly different plot data
    pd2 =  ArrayPlotData(index = x2)
    for i in range(5):
        pd2.set_data("y" + str(i), 2*jn(i,x2))

    # Create some line plots of some of the data
    plot1 = Plot(pd)
    plot1.plot(("index", "y0", "y1", "y2"), name="j_n, n<3", color="red")

    # Tweak some of the plot properties
    plot1.title = "My First Line Plot"
    plot1.padding = 50
    plot1.padding_top = 75
    plot1.legend.visible = True

    plot2 = Plot(pd2)
    plot2.plot(("index", "y0", "y1"), name="j_n, n<3", color="green")

    plot1.add(plot2)
    # Attach some tools to the plot
    broadcaster = BroadcasterTool()
    broadcaster.tools.append(PanTool(plot1))
    broadcaster.tools.append(PanTool(plot2))

    for c in (plot1, plot2):
        zoom = ZoomTool(component=c, tool_mode="box", always_on=False)
        broadcaster.tools.append(zoom)

    plot1.tools.append(broadcaster)

    return plot1

# Attributes to use for the plot view.
size=(900,500)
title="Multi-Y plot"

# # Demo class that is used by the demo.py application.
#===============================================================================
class Demo(HasTraits):
    plot = Instance(Component)

    traits_view = View(
                    Group(
                        Item('plot', editor=ComponentEditor(size=size),
                             show_label=False),
                        orientation = "vertical"),
                    resizable=True, title=title,
                    width=size[0], height=size[1]
                    )

    def _plot_default(self):
         return _create_plot_component()

demo = Demo()
if __name__ == "__main__":
    demo.configure_traits()

1 个答案:

答案 0 :(得分:2)

Chaco中的一个瑕疵(实际上是许多绘图库)是术语的重载 - 特别是“情节”这个词。

你正在创造两个不同的(大写 - “P”)Plot,但(我相信)你真的只想要一个。 Plot是容纳所有单独行... umm ...的容器。 Plot.plot方法返回LinePlot个实例的列表(此“情节”有时也称为“渲染器”)。该渲染器是您要添加到(大写 - “P”)Plot容器中的内容。 plot方法实际上创建了LinePlot实例,并将其添加到Plot容器中。 (是的,这是“plot”的三种不同用法:容器,渲染器和容器上添加/返回渲染器的方法。)

这是一个更简单的_create_plot_component版本,大致可以满足您的需求。请注意,只创建了一个(大写 - “P”)Plot容器。

def _create_plot_component():
    # Create some x-y data series to plot
    x = linspace(-2.0, 10.0, 100)
    x2 =linspace(-5.0, 10.0, 100)

    pd = ArrayPlotData(x=x, x2=x2)
    for i in range(3):
        pd.set_data("y" + str(i), jn(i,x))

    # slightly different plot data
    for i in range(3, 5):
        pd.set_data("y" + str(i), 2*jn(i,x2))

    # Create some line plots of some of the data
    canvas = Plot(pd)
    canvas.plot(("x", "y0", "y1", "y2"), name="plot 1", color="red")
    canvas.plot(("x2", "y3", "y4"), name="plot 2", color="green")
    return canvas

编辑:早期的回复通过两行修改解决了问题,但这不是解决问题的理想方法。