Question

我有一个在iPython中使用的模块。

我希望用户输入制作绘图x，y，标签，线宽等所需的所有内容。

因此用户可能会这样做：

In[1] import this_script
In[2] x=range(0,10)
In[3] y=x
In[4] magically_exposed_function plot(x,y,'r+', linewidth=2)

这意味着我的函数得到字符串图（x，y，'r +'，linewidth = 2）。这可以解析和使用ip.user_ns在iPython命名空间中找到的x和y的值，但我仍然坚持如何处理'r +'和linewidth = 2。理想情况下，我希望能够：

a）导入整个iPython名称空间，以便我可以使用x和y值

b）将整个字符串抛入plot（）

至于b），有类似的东西：

plot_string = x, y, 'r+', linewidth = 2
plot(plot_string)

会很理想，但这不会如上所示。

这可以做到这两件事吗？有更优雅的解决方案吗？

用户是否可以绘制图（x，y），我的代码可以抓住该图并编辑它？

非常感谢任何有关如何处理这种情况的建议：）

谢谢！ --Erin

[编辑]我希望能够做的演示：

import matplotlib
import wx
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigCanv
from matplotlib.figure import Figure
import IPython.ipapi
ip = IPython.ipapi.get()
import sys

class WrapperExample(wx.Frame):
    def __init__(self, *args, **kwargs):
        wx.Frame.__init__(self, None, -1)
        self.figure = Figure()
        self.axes = self.figure.add_subplot(111)
        self.axes.plot(*args, **kwargs)
        self.canvas = FigCanv(self, -1, self.figure)

def run_me(*args, **kwargs):
    """ Plot graph from iPython
    Example:
    In[1] import script
    In[2] x=range(0,10)
    In[3] y=x
    In[4] run_me x y
    """
    app = wx.PySimpleApp()
    wrap = WrapperExample(*args, **kwargs)
    wrap.Show()
    app.MainLoop()

ip.expose_magic("run_me", run_me)

[编辑]以下是我最终使用下面建议的包装器的方法：

import wx
import matplotlib
from pylab import *
import IPython.ipapi
ip = IPython.ipapi.get()

class MainCanvas(wx.Frame):
    def __init__(self, *args):
        self.figure = plt.figure()
        self.axes = self.figure.add_subplot(111)
        self.axes.plot(*args)
        show()


def run_this_plot(self, arg_s=''):
    """ Run
    Examples
    In [1]: import demo
    In [2]: rtp x y <z> 
    Where x, y, and z are numbers of any type
    """
    args = []
    for arg in arg_s.split():
        try:
            args.append(self.shell.user_ns[arg])
        except KeyError:
            raise ValueError("Invalid argument: %r" % arg)
    mc = MainCanvas(*args)

# Activate the extension
ip.expose_magic("rtp", run_this_plot)

Answer 1

解析实际字符串最好留给python。也许你想创建一个包装器：

real_plot = plot
def my_plot(*args, **kwargs):
    x, y = args[0], args[1]
    ...your extra code here...
    real_plot(*args, **kwargs)
plot = my_plot

获取用户输入以在iPython模块中绘图的更好方法是什么？

1 个答案: