将部分文件名作为函数

Question

我正在编写一个函数来导入.dta文件，这些文件将具有名称结构但具有不同的前缀。如何在字符串文件路径名中引用此前缀？例如，像

 def import_stata(code):
    df=pd.read_stata("code_dataset.dta")
    return df

其中code是要更改的前缀。现在我收到一条错误，说code_dataset不存在。

谢谢！

Answer 1

使用以下字符串连接：

def import_stata(code):
  df=pd.read_stata(code+"_dataset.dta")
  return df