Python中Stata宏的等价物答案

【问题标题】：Equivalent of Stata macros in PythonPython中Stata宏的等价物
【发布时间】：2018-07-31 18:29:50
【问题描述】：

我正在尝试使用 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 的宏类似的东西？

【问题讨论】：

Stata 中的字符串宏从外面看有点古怪。像许多其他语言一样，Stata 是基于对以前语言的折衷借鉴和一些原创思想。在这种情况下，如果我理解正确的话，灵感来自于 C 和 Unix shell 编程的混合。
你对我如何在 python 中实现类似的东西有什么想法吗？
看我的回答。抱歉花了这么长时间。
感谢您的详细说明，我现在已将接受的答案更改为您的。我肯定会研究 Pyexpander，我最终使用了你在这里建议的列表。

标签： python pandas stata statsmodels stata-macros

【解决方案1】：

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

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

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

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

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

【讨论】：