python保存变量_Python保存变量到文件

简介

数据处理过程中，中间数据的保存很重要。Linux下的工具包通常以文件为输入和输出，不存在这一问题，而对于使用Jupyter notebook/lab为主要工具的人来说，可能会忽视这一问题。可能后果是重新花费计算的时间和精力。

本文将介绍几种Python中常用的保存内存中变量的方式，供参考。

pickle

pickle是Python官方自带的库，提供dump函数实现Python对象的保存。支持自定义的对象，非常方便。Pandas的DataFrame和Obspy的Stream也都可以保存成pickle的格式，不知道是否利用了该库。支持自定义对象

操作方便（不需要特意将变量变换为某种特定格式）

使用方法为

import pickle

# a = ...

pickle.dump(a, open('a.pkl', 'wb'))

SciPy (savemat)

scipy提供的savemat函数是我常用的保存数据的方式支持numpy array

可以与MATLAB实现数据交换

使用方法为

from scipy.io import savemat

mdict = {'a': a}

savemat('a.mat', mdict)

json （待补充）

json是另一个官方库，支持列表和字典。文本文件，可读性强

格式通用

Pandas (待补充）

pandas提供的DateFrame可以实现表格数据的保存。