软件测试|使用Python读取CSV文件的详细指南

简介

CSV（Comma-Separated Values）是一种常用的数据存储格式，它以逗号为分隔符将数据以文本形式存储在文件中。Python是一门强大的编程语言，提供了许多库和工具，使得读取和处理CSV文件变得非常简便。在本文中，我们将深入介绍如何使用Python读取CSV文件的详细步骤。

步骤1：导入所需的库

在开始之前，需要导入Python中处理CSV文件所需的库。Python标准库中的csv模块是一个处理CSV文件的良好选择。

import csv

步骤2：打开CSV文件

在读取CSV文件之前，需要使用Python的内置open函数打开文件。确保提供正确的文件路径，并指定文件的打开模式为读取（'r'）。

file_path = 'your_file.csv'

with open(file_path, 'r') as csv_file:
    # 后续操作将在此代码块中进行

步骤3：创建CSV读取器

在打开文件后，需要创建一个CSV读取器对象，用于我们逐行读取CSV文件的内容。

with open(file_path, 'r') as csv_file:
    csv_reader = csv.reader(csv_file)
    
    for row in csv_reader:
        # 每次迭代将读取一行数据并存储在'row'变量中

步骤4：处理CSV数据

现在我们可以通过迭代CSV读取器对象来访问每一行数据。每一行数据都被解析为一个由字段组成的列表。

with open(file_path, 'r') as csv_file:
    csv_reader = csv.reader(csv_file)
    
    for row in csv_reader:
        # 您可以通过索引访问每个字段
        # 例如：第一个字段 row[0], 第二个字段 row[1], 依此类推
        
        # 进行您的数据处理操作，例如打印每一行数据
        print(row)

步骤5：使用CSV DictReader

如果CSV文件的第一行包含列标题，我们还可以使用csv.DictReader来创建一个能够返回每行数据作为字典的读取器。这样做使得我们可以通过列标题访问数据，使得代码更加清晰易读。

with open(file_path, 'r') as csv_file:
    csv_reader = csv.DictReader(csv_file)
    
    for row in csv_reader:
        # 可以通过列标题访问每个字段
        # 例如：row['Name'], 依此类推
        
        # 进行数据处理操作，例如打印特定字段的值
        print(row['Name'])

使用示例

假设我们有一个CSV文件，内容如下：

name, id, major
muller, 01, math
salah, 02, music
messi, 03, english

我们要完整读取其内容，代码如下：

import csv
# open file by passing the file path.
with open('files/data.csv', 'r') as csv_file:
    csv_read = csv.reader(csv_file, delimiter=',')  #Delimeter is comma
    count_line = 0
    # Iterate the file object or each row of the file
    for row in csv_read:
        if count_line == 0:
            print(f'Column names are {", ".join(row)}')
            count_line += 1
        else:
            print(f'\t{row[0]} id is:  {row[1]} and major is: {row[2]}.')
            count_line += 1
    print(f'Processed {count_line} lines.')

运行脚本，结果如下：

Column names are name,  id,  major
	muller id is:   01 and major is:  math.
	salah id is:   02 and major is:  music.
	messi id is:   03 and major is:  english.
Processed 4 lines.

总结

通过按照上述步骤，我们可以轻松地使用Python读取和处理CSV文件中的数据。无论是简单的数据分析还是复杂的数据处理，Python的CSV处理功能都能满足我们的需求。

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：

这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴上万个测试工程师们走过最艰难的路程，希望也能帮助到你！