python - 如何将 Pandas 数据框保存到 sav 文件中

Question

我有以下数据框:

import pandas as pd
mc_all = pd.DataFrame({'col_1': [0,1,1,2],
'col_2' : ['france','france','uk','uk']})

我尝试这样做是为了将这个dataframe 保存到sav 文件中

args = ( list(mc_all.columns), dict(zip(list(mc_all.columns),[0]*len(list(mc_all.columns)))) )
with SavWriter('mc_all.sav',*args) as writer:
    writer.writerows(mc_all)

根据 this但它不起作用。它抛出一个错误:

error.SPSSIOError: Problem committing header [SPSS_INVALID_HANDLE]

有什么想法吗？

Answer 1

您可以使用 pyreadstat:

import pyreadstat

pyreadstat.write_sav(mc_all, savFileName)

更多信息在这里:

https://github.com/Roche/pyreadstat#writing-files