欢迎访问宙启技术站
智能推送

利用save()函数将Python程序的结果保存为CSV文件的方法

发布时间:2023-12-18 22:17:37

在Python中,可以使用csv模块来将程序的结果保存为CSV文件。csv模块提供了很多用于读写CSV文件的函数和类。

首先,我们需要导入csv模块:

import csv

然后,我们可以使用csv.writer()函数创建一个写入CSV文件的对象。这个函数接受一个文件对象作为参数,我们可以使用open()函数来创建一个文件对象。例如,我们要将结果保存到名为output.csv的文件中:

output_file = open("output.csv", "w", newline="")
csv_writer = csv.writer(output_file)

在这里,open()函数使用"w"参数来表示写入模式打开文件,newline=""参数用于避免在Windows系统中产生额外的换行符。

接下来,我们可以使用csv_writer.writerow()函数来写入一行数据。这个函数接受一个列表作为参数,列表中的每个元素将被写入CSV文件的一列。例如,我们要写入一行包含"Name""Age"的数据:

csv_writer.writerow(["Name", "Age"])

接着,我们可以使用csv_writer.writerows()函数来写入多行数据。这个函数接受一个嵌套列表作为参数,嵌套列表中的每个子列表将被写入CSV文件的一行。例如,我们要写入多行数据:

data = [
    ["John", 25],
    ["Alice", 30],
    ["Bob", 35]
]

csv_writer.writerows(data)

最后,我们需要关闭文件对象以确保写入的数据被保存到文件中:

output_file.close()

下面是一个完整的例子,该例子将一些学生的姓名和年龄保存为CSV文件:

import csv

data = [
    ["John", 25],
    ["Alice", 30],
    ["Bob", 35]
]

output_file = open("output.csv", "w", newline="")
csv_writer = csv.writer(output_file)

csv_writer.writerow(["Name", "Age"])
csv_writer.writerows(data)

output_file.close()

运行这个程序后,将会在当前目录下生成一个output.csv文件,其中包含了保存的数据。