温馨提示×

python怎么查找csv中某个数据

小亿
258
2023-11-01 09:21:16
栏目: 编程语言

要查找CSV文件中的某个数据,可以使用Python的csv模块读取CSV文件,并使用循环遍历文件中的每一行,然后逐行查找目标数据。以下是一个示例代码:

import csv

def find_data(csv_file, target_data):
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            if target_data in row:
                return row
    return None

# 示例用法
csv_file = 'data.csv'
target_data = 'John'
result = find_data(csv_file, target_data)
if result:
    print("找到数据:", result)
else:
    print("未找到数据")

在上面的示例中,我们定义了一个find_data函数,该函数接受两个参数:CSV文件路径和目标数据。函数使用with open语句打开CSV文件,并使用csv.reader创建一个读取器对象。然后,通过一个循环遍历每一行,如果目标数据在当前行中,则返回该行数据。如果循环结束后仍未找到目标数据,则返回None

示例用法中,我们传入CSV文件路径和要查找的数据,然后检查返回结果是否为None,如果不是,则打印找到的数据。

0