如何在python代码中指定保存的文件格式

python

python指定保存文件格式的方法：

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

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

若文件过大

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

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')

更多Python知识请关注云海天Python教程栏目。