Python 读取和保存数据：CSV、Excel 文件操作指南

读取数据：

使用 Python 自带的 csv 模块读取 csv 文件

import csv

with open('file.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

使用 pandas 库读取 csv 文件

import pandas as pd

data = pd.read_csv('file.csv')
print(data)

使用 xlrd 库读取 excel 文件

import xlrd

workbook = xlrd.open_workbook('file.xlsx')
worksheet = workbook.sheet_by_index(0)

for row_index in range(worksheet.nrows):
    row = worksheet.row(row_index)
    print(row)

保存数据：

使用 Python 自带的 csv 模块保存数据到 csv 文件

import csv

data = [['name', 'age'], ['Alice', 25], ['Bob', 30], ['Charlie', 35]]

with open('file.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

使用 pandas 库保存数据到 csv 文件

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Charlie'], 'age': [25, 30, 35]}
df = pd.DataFrame(data)

df.to_csv('file.csv', index=False)

使用 xlwt 库保存数据到 excel 文件

import xlwt

data = [['name', 'age'], ['Alice', 25], ['Bob', 30], ['Charlie', 35]]

workbook = xlwt.Workbook()
worksheet = workbook.add_sheet('Sheet1')

for row_index, row_data in enumerate(data):
    for col_index, col_data in enumerate(row_data):
        worksheet.write(row_index, col_index, col_data)

workbook.save('file.xls')