pandas库pd.read_pickle操作读取pickle数据与.to_pickle()永久储存数据

"""
【python】pandas库pd.read_pickle操作读取pickle数据与.to_pickle()永久储存数据

"""

# 1.pickle.dump(obj, file, [,protocol])
import pickle
# 写入一个文件，用二进制的形式
f = open('data_one.pkl', 'wb')
# 等待写入的数据
datas = {'name': 'oscar', 'age': 25, 'high': 175}
# dump函数将obj数据datas导入到file中
data_one = pickle.dump(datas, f, -1)
# f文件结束操作句柄
f.close()


# 2.pickle.load(file)
# 打开一个data_one文件
f = open('data_one.pkl', 'rb')
# 使用load方法将数据从pkl文件中读取出来
print(pickle.load(f))
f.close()


# 3.使用pandas库进行pickle更加简单
# 使用DataFrame的to_pickle属性就可以生成pickle文件对数据进行永久储存
import pandas as pd
import numpy as np
df = pd.DataFrame(np.arange(20).reshape(4, 5))
print('写入数据\n',df)
# 使用DataFrame的to_pickle属性就可以生成pickle文件对数据进行永久储存
df.to_pickle('foo.pkl')
# 使用pandas库的pd.read_pickle读取pickle数据
dr = pd.read_pickle('foo.pkl')
print('读取数据\n', dr)


# 4.删除文件
import os
os.remove('data_one.pkl')
os.remove('foo.pkl')