[教程]Python存储表格：轻松掌握CSV、Excel与数据库三大存储方法

引言在Python中，处理和存储表格数据是数据分析、数据科学和软件开发中常见的需求。CSV、Excel和数据库是三种常用的数据存储格式，它们各自具有不同的特点和适用场景。本文将详细介绍如何在Pytho...

引言

在Python中，处理和存储表格数据是数据分析、数据科学和软件开发中常见的需求。CSV、Excel和数据库是三种常用的数据存储格式，它们各自具有不同的特点和适用场景。本文将详细介绍如何在Python中使用这些方法来存储表格数据。

CSV文件存储

CSV（Comma-Separated Values，逗号分隔值）是一种简单的文本格式，常用于存储表格数据。Python的内置库csv提供了读取和写入CSV文件的功能。

写入CSV文件

以下是一个示例代码，演示如何使用csv库将数据写入CSV文件：

import csv
# 数据列表
data = [ ['id', 'name', 'age'], [1, 'Alice', 25], [2, 'Bob', 30], [3, 'Charlie', 35]
]
# 写入CSV文件
with open('data.csv', 'w', newline='') as csvfile: writer = csv.writer(csvfile) writer.writerows(data)

读取CSV文件

以下是一个示例代码，演示如何使用csv库读取CSV文件：

import csv
# 读取CSV文件
with open('data.csv', 'r') as csvfile: reader = csv.reader(csvfile) for row in reader: print(row)

Excel文件存储

Excel是一种电子表格格式，广泛用于数据存储和展示。Python的库openpyxl和xlrd提供了读取和写入Excel文件的功能。

写入Excel文件

以下是一个示例代码，演示如何使用openpyxl库将数据写入Excel文件：

from openpyxl import Workbook
# 创建工作簿和工作表
wb = Workbook()
ws = wb.active
# 写入数据
ws.append(['id', 'name', 'age'])
ws.append([1, 'Alice', 25])
ws.append([2, 'Bob', 30])
ws.append([3, 'Charlie', 35])
# 保存工作簿
wb.save('data.xlsx')

读取Excel文件

以下是一个示例代码，演示如何使用openpyxl库读取Excel文件：

from openpyxl import load_workbook
# 加载工作簿
wb = load_workbook('data.xlsx')
# 获取工作表
ws = wb.active
# 读取数据
for row in ws.iter_rows(values_only=True): print(row)

数据库存储

数据库是一种用于存储、管理和检索数据的系统。Python的库sqlite3提供了对SQLite数据库的支持。

创建数据库和表

以下是一个示例代码，演示如何使用sqlite3库创建数据库和表：

import sqlite3
# 创建数据库连接
conn = sqlite3.connect('data.db')
# 创建游标对象
cursor = conn.cursor()
# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS users ( id INTEGER PRIMARY KEY, name TEXT NOT NULL, age INTEGER NOT NULL
)
''')
# 提交事务
conn.commit()
# 关闭连接
conn.close()

插入数据

以下是一个示例代码，演示如何使用sqlite3库插入数据：

import sqlite3
# 创建数据库连接
conn = sqlite3.connect('data.db')
# 创建游标对象
cursor = conn.cursor()
# 插入数据
cursor.execute('''
INSERT INTO users (name, age) VALUES (?, ?)
''', ('Alice', 25), ('Bob', 30), ('Charlie', 35))
# 提交事务
conn.commit()
# 关闭连接
conn.close()

总结

本文介绍了Python中三种常用的表格数据存储方法：CSV、Excel和数据库。通过学习这些方法，您可以轻松地将数据存储为不同的格式，以满足不同的需求。在实际应用中，您可以根据具体场景选择最合适的方法。