如何保存python文件

hakesashou2024-11-19 9:10

1、保存为二进制文件，pkl格式

复制代码

import pickle
pickle.dump(data,open('file_path','wb'))  #后缀.pkl可加可不加

若文件过大

复制代码

pickle.dump(data,open('file_path', 'wb'),protocol=4)

读取该文件：

复制代码

data= pickle.load(open('file_path','rb'))

2、保存为二进制文件，npz格式

复制代码

import numpy as np
np.savez('file_path/file_name.npz', data1=X,data2=y)

读取该文件：

复制代码

with np.load('file_path/file_name.npz') as data:
    X = data['data1']
    y = data['data2']

3、DataFrame文件保存为.csv

复制代码

dataframe_file.to_csv("file_path/file_name.csv", index=False)

读取该文件：

复制代码

import pandas as pd
df = pd.read_csv('file_path/file_name.csv')