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

使用Python编写的文件操作程序:读取和写入CSV文件

发布时间:2023-12-04 09:38:38

Python是一种强大的编程语言,提供了许多用于文件操作的库和函数。其中一个常见的文件格式是CSV(逗号分隔值),它是一种用逗号将数据值分隔为不同列的简单文本文件格式。

在Python中,我们可以使用内置的csv模块来读取和写入CSV文件。以下是一个简单的Python程序,演示了如何使用csv模块读取和写入CSV文件:

import csv

# 1. 读取CSV文件
def read_csv_file(file_name):
    with open(file_name, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            print(row)  # 打印每行数据

# 2. 写入CSV文件
def write_csv_file(file_name, data):
    with open(file_name, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(data)

# 3. 使用例子
def example_usage():
    # 读取CSV文件
    read_csv_file('data.csv')

    # 写入CSV文件
    data = [
        ['Name', 'Age', 'Country'],
        ['John', 25, 'USA'],
        ['Emma', 30, 'UK'],
        ['Daniel', 28, 'Canada']
    ]
    write_csv_file('output.csv', data)

# 主函数
if __name__ == '__main__':
    example_usage()

在上面的示例中,read_csv_file函数演示了如何打开CSV文件并使用csv.reader来读取文件内容。每一行都表示为一个列表,我们可以使用print语句打印每一行。

write_csv_file函数展示了如何打开一个新的CSV文件并使用csv.write来将数据写入文件。数据被组织成一个二维列表,其中每个内部列表表示一行数据。

在使用例子中,我们首先使用 个函数读取名为data.csv的CSV文件,并将其打印出来。然后,我们创建一个包含一些数据的二维列表,并使用第二个函数将该数据写入名为output.csv的新CSV文件。

这只是一个基本的文件操作示例,你可以根据需要进行进一步的处理和操作。CSV文件格式非常灵活,可以存储各种类型的数据,并且常用于数据导入和导出。在实际应用中,你可能需要更复杂的数据处理和操作,例如过滤、排序和计算数据等。Python提供了强大的库和函数来处理这些操作,所以你可以根据需要自由地扩展这个程序。