c语言编程笔录

首页 >   > 笔记大全

笔记大全

python创建cvs文件 python cvs库

更新时间:2023-10-17

什么是CSV文件?

CSV文件是一种用于存储和传输数据的简单格式。它由以逗号作为分隔符的数据行组成,每行数据可以映射为一行表格中的单元格。CSV文件可以被许多应用程序轻松地解析和生成,因为它是一个简单的纯文本格式。

# 示例:创建一个CSV文件

import csv

header = ['姓名', '年龄', '性别']
data = [['aaa', '21', '男'], ['bbb', '22', '女'], ['ccc', '23', '男']]

with open('data.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(header)
    writer.writerows(data)

Python CSV库

Python内置了用于读写CSV文件的csv模块。CSV模块提供了一些针对CSV文件的特定功能,例如自动拆分行和字段以及在写入时自动添加逗号分隔符。这些函数可用于打开、读取和写入CSV文件。下面是使用Python CSV库读取和写入csv文件的例子:

# 示例:读取CSV文件

import csv

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)
    for row in reader:
        print(row)
        
# 示例:写入CSV文件

import csv

header = ['姓名', '年龄', '性别']
data = [['aaa', '21', '男'], ['bbb', '22', '女'], ['ccc', '23', '男']]

with open('data_new.csv','w') as f:
    writer = csv.writer(f)
    writer.writerow(header)
    writer.writerows(data)

CSV文件与Pandas

Pandas库也提供了读取和写入CSV文件的函数。这个库可以很好地扩展CSV文件的读取和写入功能。Pandas库可以让您方便地将CSV文件读入DataFrame对象中,并轻松地进行数据清洗和分析。

# 示例:读取CSV文件到DataFrame

import pandas as pd

df = pd.read_csv('data.csv')

# 示例:将DataFrame写入CSV文件

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

总结

CSV文件是一种流行的简单数据格式,它被广泛用于数据交换和存储。Python内置了用于读写CSV文件的模块,其中csv模块是最简单的方法,而pandas库则提供了更多的扩展功能。无论你是初学者还是资深开发者,Python都提供了足够的支持和工具,帮助你轻松地管理CSV文件。