python - Python中Stata宏的等价物

Question

我正在尝试使用 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)

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

• How do I create a stata local macro in python?

对于R还有另一个问题:

• How do I create a "macro" for regressors in R?

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

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

Answer 1

看起来您只需要 * 和 ** 运算符来调用函数:

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

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

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

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