Question

我想做以下事情：

In[1]: name = 'long_name_to_type_every_now_and_then.py'

In[2]: %run name

但实际上这会尝试运行'name.py'，这不是我想要做的。

是否有将变量转换为字符串的一般方法？

如下所示：

In[3]: %run %name%

Answer 1

IPython使用$name扩展变量，bash-style。这对于所有魔法都是如此，而不仅仅是%run。

所以你会这样做：

In [1]: filename = "myscript.py"

In [2]: %run $filename
['myscript.py']

myscript.py包含：

import sys
print(sys.argv)

通过Python的花式字符串格式，您甚至可以将表达式放在{}：

中

In [3]: args = ["arg1", "arg2"]

In [4]: %run $filename {args[0]} {args[1][-2:]}
['myscript.py', 'arg1', 'g2']

Answer 2

使用get_ipython()获取对当前InteractiveShell的引用，然后调用magic()方法：

In [1]: ipy = get_ipython()

In [2]: ipy.magic("run foo.py")
ERROR: File `u'foo.py'` not found.

修改请参阅minrk's answer - 这是一种更好的方法。

Answer 3

内置%run魔术功能似乎无法实现这一点。你的问题让我陷入了一个兔子洞，我想看看做类似的事情是多么容易。最后，为了创建另一个仅使用execfile()的魔术函数，所有这些努力似乎毫无意义。也许这会对某人有所帮助。

# custom_magics.py
from IPython.core.magic import register_line_magic, magics_class, line_magic, Magics

@magics_class
class StatefulMagics(Magics):
    def __init__(self, shell, data):
        super(StatefulMagics, self).__init__(shell)
        self.namespace = data

    @line_magic
    def my_run(self, line):
        if line[0] != "%":
            return "Not a variable in namespace"
        else:
            filename = self.namespace[line[1:]].split('.')[0]
            filename += ".py"
            execfile(filename)
        return line

class Macro(object):
    def __init__(self, name, value):
        self.name = name
        self._value = value
        ip = get_ipython()
        magics = StatefulMagics(ip, {name: value})
        ip.register_magics(magics)

    def value(self):
        return self._value

    def __repr__(self):
        return self.name

使用这对类（并给出一个python脚本tester.py），可以使用新创建的“my_run”魔术函数创建和使用“宏”变量，如下所示：

In [1]: from custom_magics import Macro

In [2]: Macro("somename", "tester.py")
Out[2]: somename

In [3]: %my_run %somename
I'm the test file and I'm running!
Out[3]: u'%somename'

是的，这是一个巨大的，可能是浪费的黑客。在那种情况下，我想知道是否有一种方法可以将绑定到Macro对象的名称用作宏的实际名称。请看看。

Answer 4

如果参数中可能有空格，例如文件名，最好使用此名称：

%run "$filename"

Answer 5

如果“变量”是指用户命名空间中可用的变量，那么我建议使用 get_ipython().ev(varname) (docs) 来获取函数中的变量：

from __future__ import print_function
import argparse
import shlex

from IPython import get_ipython
from IPython.core.magic import register_cell_magic

@register_cell_magic
def magical(line, cell):
    parser = argparse.ArgumentParser()
    parser.add_argument("varname")
    # using shlex to split the line in a sys.argv-like format
    args = parser.parse_args(shlex.split(line))
    ipython = get_ipython()

    v = ipython.ev(args.varname)
    # do whatever you want here
    print(v)

上面的这个模板对我很有用 - 我用它来制作一个 jinja2 单元魔法，只要单元运行，它也会打印到文件中。

如何在IPython中将变量传递给magic'run'函数

5 个答案: