首页 > 脚本专栏 > python > Python保存数据到文件

Python保存数据到文件的实现方式

2023-11-02 10:34:46 作者：Just_Paranoid

这篇文章主要介绍了Python保存数据到文件的实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

IO（Input/Output）是指计算机与外部环境之间的数据交换。Python中的IO操作分为文件IO和网络IO两种。

文件IO：Python提供了内置的文件对象来实现文件IO操作。文件对象是通过内置的open()函数创建的，它可以用于读取、写入、修改文件。

网络IO：Python中通过socket模块支持网络IO操作。使用socket模块，可以创建TCP/UDP套接字来实现网络通信。

Python中还提供了许多高级IO库，如：io、pickle、json等，它们能够方便地对数据进行序列化和反序列化。

此外，Python还支持标准输入输出（stdin/stdout），可以与操作系统进行交互，实现命令行输入输出操作。

方法一：open函数

open函数是Python内置函数之一，用于打开文件并返回文件对象，可以用于读取或写入文件。

它的常用语法是：

open(file, mode='r', buffering=-1, encoding=None, errors=None, newline=None, closefd=True, opener=None)

参数解析：

file表示要打开的文件名（包括路径），
mode表示打开文件的模式，默认为只读模式（‘r’）。对于写入模式，可以使用’w’、‘x’、'a’等方式，分别表示覆盖写入、创建新文件并写入、在文件末尾添加内容等。
buffering表示缓冲策略，默认为使用系统默认缓冲机制，
encoding表示文件编码方式，
errors表示编码错误时的处理方式。
newline表示换行符的转换策略，
closefd表示是否在文件关闭时同时关闭文件描述符，
opener表示自定义开启文件的方式。

#使用with open()新建对象f
with open('file.txt','w',encoding='utf-8') as f:
    for i in comments:
        #写入数据，文件保存在上面指定的目录，加\n为了换行更方便阅读  
        f.write(i+'\n')

方法二：numpy

NumPy是Python科学计算中最重要的模块之一。

它提供了一个非常灵活的数组对象，可以方便地处理数值数据。

NumPy还提供了高效的矩阵运算、快速的傅里叶变换等等。

# 导入NumPy模块
import numpy as np

# 创建NumPy数组对象并填充数据
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 调用NumPy提供的savetxt()函数将数组数据写入文件中
np.savetxt('output.txt', arr)
# savetxt()函数有很多可选参数来控制格式、精度等选项。例如，可以使用delimiter参数设置分隔符，使用fmt参数设置输出格式
np.savetxt('output.txt', arr, delimiter=',', fmt='%d')

方法三：csv

Python的csv模块是一个用于读写CSV（Comma-Separated Values）格式文件的标准库。

CSV格式常用于大量数据的存储和交换，例如计算机数据、表格数据、统计数据等等。

import csv

# 打开文件
with open('data.csv', 'w', newline='') as file:
    # 创建writer对象
    writer = csv.writer(file)
    
    # 写入一行数据
    writer.writerow(['姓名', '年龄', '性别'])
    
    # 写入多行数据
    rows = [
        ['张三', 20, '男'],
        ['李四', 22, '女'],
        ['王五', 18, '男'],
    ]
    writer.writerows(rows)

方法四：DataFrame

pandas是一个用于数据分析的Python库，其中最主要的数据结构是DataFrame。

DataFrame类中提供了多个方法用于写入数据到文件中， to_csv()将DataFrame写入CSV文件、to_excel()DataFrame写入Excel文件、to_json()方法将DataFrame转换为JSON格式数据，这些方法均提供多个参数可供使用，如控制输出格式、数据类型、缺失值处理等，详见pandas官方文档。

to_csv()方法：将DataFrame写入CSV文件

import pandas as pd

# 创建一个DataFrame
data = {'name':['Tom', 'Jack', 'Steve', 'Ricky'], 'age':[28, 34, 29, 42]}
df = pd.DataFrame(data)

# 将DataFrame写入CSV文件
df.to_csv('output.csv', index=False)

to_excel()方法：将DataFrame写入Excel文件

import pandas as pd

# 创建一个DataFrame
data = {'name':['Tom', 'Jack', 'Steve', 'Ricky'], 'age':[28, 34, 29, 42]}
df = pd.DataFrame(data)

# 将DataFrame写入Excel文件
df.to_excel('output.xlsx', index=False)

方法五：codecs

Python中的codecs模块提供了一些编码和解码功能，可以帮助我们在读写文件时处理不同字符集的问题。

codecs模块还提供了许多其他的函数，如open()、encode()、decode()、lookup()等等，可用于处理各种常见的编码方式，如UTF-8、ASCII等等。

import codecs #或者io，使用哪种包无所谓

# 代码使用`codecs.open()`函数打开文件，第一个参数是文件路径，第二个参数指定文件打开模式，`w`表示写入，`r`表示读取。第三个参数是指定文件编码，这里是UTF-8编码。
with codecs.open('your_file.txt', 'r', 'utf-8') as f:
    f.write('This method is prior')

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。