使用pandas库读取和写入csv文件的实现示例

import csv

# 打开 CSV 文件并读取内容
with open('data.csv', mode='r', encoding='utf-8') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)  # 输出每一行的数据

import csv

# 准备数据
data = [['Name', 'Age'], ['Alice', 25], ['Bob', 30]]

# 写入 CSV 文件
with open('output.csv', mode='w', encoding='utf-8', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

import pandas as pd

# 读取 CSV 文件
df = pd.read_csv('data.csv')
print(df.head())  # 显示前几行数据

# 创建一个新的 DataFrame
new_df = pd.DataFrame({
    'Name': ['Charlie', 'David'],
    'Age': [35, 40]
})

# 将数据写入 CSV 文件
new_df.to_csv('output_pandas.csv', index=False)

import timeit

# 测试 csv 模块的读取速度
csv_time = timeit.timeit(
    "list(csv.reader(open('large_data.csv', 'r')))[:1000]",
    setup="import csv",
    number=1
)

# 测试 Pandas 的读取速度
pandas_time = timeit.timeit(
    "pd.read_csv('large_data.csv').head(1000)",
    setup="import pandas as pd",
    number=1
)

print(f"CSV 模块耗时: {csv_time}")
print(f"Pandas 耗时: {pandas_time}")