将列表保存到文件并将id加载到Pandas上的最佳方法是什么？

Question

如何像这样保存和加载列表，以便我可以在其他地方加载

Index(['province_BANTEN', 'province_DKI JAKARTA', 'province_JAWA BARAT',
       ...
       'lending_count_l3', 'lending_cumcount_l2', 'lending_cumcount_l3'],
      dtype='object', length=341)

我试图保存到csvdf.to_csv("model_column.csv")，但是AttributeError: 'Index' object has no attribute 'to_csv'

有没有更好的办法来挽救这一切

Answer 1

您可以使用pickle模块保存数据帧，例如：

df = pd.DataFrame()

# save dataframe as pickle
with open('model_columns.pickle', 'wb') as f:
    pickle.dump(df, f, protocol=pickle.HIGHEST_PROTOCOL)

# load dataframe as pickle
with open('model_columns.pickle', 'rb') as f:
    new_df = pickle.load(f)

pickle可以将Python（如dataframe）对象结构转换为字节流并存储

Answer 2

可能会构造一个新的数据帧：

df = pd.DataFrame(index=df)
df.to_csv("model_column.csv")