相当于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)

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

• 如何在python中创建stata本地宏

R还有另一个问题：

• 我如何创建一个"；宏"；对于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)