2

我正在尝试使用 Python 进行统计分析。

在 Stata 中，我可以定义本地宏并根据需要扩展它们：

program define reg2
    syntax varlist(min=1 max=1), indepvars(string) results(string)
    if "`results'" == "y" {
        reg `varlist' `indepvars'
    }
    if "`results'" == "n" {
        qui reg `varlist' `indepvars'
    }
end

sysuse auto, clear

所以而不是：

reg2 mpg, indepvars("weight foreign price") results("y")

我可以：

local options , indepvars(weight foreign price) results(y) 
reg2 mpg `options'

甚至：

local vars weight foreign price
local options , indepvars(`vars') results(y) 
reg2 mpg `options'

Stata 中的宏帮助我编写干净的脚本，而无需重复代码。

在 Python 中，我尝试了字符串插值，但这在函数中不起作用。

例如：

def reg2(depvar, indepvars, results):
    print(depvar)
    print(indepvars)
    print(results)

以下运行良好：

reg2('mpg', 'weight foreign price', 'y')

但是，这两个都失败了：

regargs = 'mpg', 'weight foreign price', 'y'
reg2(regargs)

regargs = 'depvar=mpg, covariates=weight foreign price, results=y'
reg2(regargs)

我发现了一个类似的问题，但它没有回答我的问题：

R还有另一个关于这个的问题：

但是，我找不到任何专门用于 Python 的东西。

我想知道 Python 中是否有与 Stata 的宏类似的东西？

4

1 回答 1

1

看起来您只需要*and**运算符来调用函数：

regargs = 'mpg', 'weight foreign price', 'y'
reg2(*regargs)

用于*将列表或元组扩展为位置参数，或用于**将字典扩展为需要它们的函数的关键字参数。

对于您的关键字示例，您需要稍微更改声明：

regargs = dict(depvar='mpg', covariates='weight foreign price', results='y')
reg2(**regargs)
于 2018-08-01T17:55:53.423 回答