在WRDS CRSP查询中使用外部文件的Python值

Question

我希望从外部文件( Python )中读取字典值(CRSP )，并将它们用于将从特定表返回行的函数中。下面的代码来自于WRDS Python支持站点，它与内联编写的PERMNO值很好地工作。

def main():
    parm = {'permnos': ('22074', '20482', '10049', '19641', '18980')}
    data = db.raw_sql('SELECT date,permno,comnam,cusip FROM crsp.dseall WHERE permno in %(permnos)s', params=parm)
    data.to_csv(r'cusip-result-2.csv')
    print(data)
if __name__=="__main__":
    main()

然而，我有541 PERMNOS，并希望避免手动进入他们的内联！

谢谢你的提示和指点！

Answer 1

首先，将每个permnos写入一个名为permnos.txt的文件，格式如下：

22074
20482
...
18980

然后，您可以执行以下操作：

def main():
    parm = {'permnos': tuple([line for line in open('permnos.txt', 'r')])}
    data = db.raw_sql('SELECT date,permno,comnam,cusip FROM crsp.dseall WHERE permno in %(permnos)s', params=parm)
    data.to_csv(r'cusip-result-2.csv')
    print(data)
if __name__=="__main__":
    main()