如何将熊猫数据框保存到 sav 文件中

Question

我有以下dataframe：

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: 提交标头问题 [SPSS_INVALID_HANDLE]

有什么想法吗？

Answer 1

你可以使用pyreadstat：

import pyreadstat

pyreadstat.write_sav(mc_all, savFileName)

更多信息在这里：

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

Answer 2

.writerows 不能作为输入。您必须将数据框转换为 np.array 才能写入。

import pandas as pd
import numpy as np

mc_all = pd.DataFrame({'col_1': [0,1,1,2],
                       'col_2' : ['france','france','uk','uk']})
savFileName = 'mc_all.sav'
args = (['col_1', 'col_2'], dict(col_1=0, col_2=0))
array = mc_all.values
with SavWriter(savFileName, *args) as writer:
    writer.writerows(array)