下面是分享Python写CSV文件的两种方法的攻略:
方法1:使用csv模块写入CSV文件
- 导入csv模块
import csv
- 创建CSV文件
假设我们要写入一个名为“data.csv”的CSV文件,我们可以通过以下代码创建它:
with open('data.csv', mode='w', newline='') as csv_file:
writer = csv.writer(csv_file)
writer.writerow(['Name', 'Age', 'Gender'])
在上述代码中,我们使用open()
函数打开一个名为“data.csv”的文件,并将打开的文件对象存储在csv_file
变量中。mode='w'
参数指示我们将以写入模式打开文件。newline=''
参数指示我们在写入CSV文件时不使用指定换行符。接下来,我们使用csv.writer()
函数创建一个写入器对象,并将其存储在writer
变量中。我们通过writerow()
函数写入CSV文件的第一行标题行。
- 写入CSV文件数据
我们可以通过以下代码将多行数据写入CSV文件:
with open('data.csv', mode='w', newline='') as csv_file:
writer = csv.writer(csv_file)
writer.writerow(['Name', 'Age', 'Gender'])
writer.writerow(['John', '25', 'Male'])
writer.writerow(['Jane', '21', 'Female'])
writer.writerow(['Bob', '40', 'Male'])
在上述代码中,我们使用writerow()
函数将三行数据写入CSV文件。每行数据都是以列表形式传递给函数。
- 读取CSV文件内容
我们可以通过以下代码读取刚才写入CSV文件的内容:
with open('data.csv', mode='r') as csv_file:
reader = csv.reader(csv_file)
for row in reader:
print(row)
在上述代码中,我们使用csv.reader()
函数创建一个读取器对象,并将其存储在reader
变量中。我们通过for
循环逐行读取CSV文件中的所有行,并打印每行的内容。
方法2:使用Pandas库写入CSV文件
- 导入Pandas库
import pandas as pd
- 创建DataFrame对象
假设我们要写入的数据如下:
Name | Age | Gender |
---|---|---|
John | 25 | Male |
Jane | 21 | Female |
Bob | 40 | Male |
我们可以通过以下代码创建一个Pandas的DataFrame对象:
data = {'Name': ['John', 'Jane', 'Bob'], 'Age': [25, 21, 40], 'Gender': ['Male', 'Female', 'Male']}
df = pd.DataFrame(data)
在上述代码中,我们以字典形式创建了三列数据,并使用它们创建了一个DataFrame对象。
- 写入CSV文件
我们可以使用to_csv()
函数将DataFrame对象写入CSV文件:
df.to_csv('data.csv', index=False)
在上述代码中,我们调用df.to_csv()
函数并将其结果存储在文件“data.csv”中,该文件将写入在与Python代码相同的目录中。index=False
参数指示我们不将行索引写入CSV文件中。
- 读取CSV文件内容
我们可以使用Pandas的read_csv()
函数读取刚才写入CSV文件的内容:
df = pd.read_csv('data.csv')
print(df)
在上述代码中,我们使用pd.read_csv()
函数读取了“data.csv”文件,并将其读取结果存储在DataFrame对象df
中。我们使用print()
函数打印DataFrame对象的内容,以查看它是否正确读取。
以上就是分享Python写CSV文件的两种方法的完整攻略和示例说明,希望能帮到您。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:分享python 写 csv 文件的两种方法 - Python技术站