溫馨提示×

python怎么查找csv中某個數(shù)據(jù)

小億
217
2023-11-01 09:21:16
欄目: 編程語言

要查找CSV文件中的某個數(shù)據(jù),可以使用Python的csv模塊讀取CSV文件,并使用循環(huán)遍歷文件中的每一行,然后逐行查找目標數(shù)據(jù)。以下是一個示例代碼:

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("找到數(shù)據(jù):", result)
else:
    print("未找到數(shù)據(jù)")

在上面的示例中,我們定義了一個find_data函數(shù),該函數(shù)接受兩個參數(shù):CSV文件路徑和目標數(shù)據(jù)。函數(shù)使用with open語句打開CSV文件,并使用csv.reader創(chuàng)建一個讀取器對象。然后,通過一個循環(huán)遍歷每一行,如果目標數(shù)據(jù)在當前行中,則返回該行數(shù)據(jù)。如果循環(huán)結束后仍未找到目標數(shù)據(jù),則返回None。

示例用法中,我們傳入CSV文件路徑和要查找的數(shù)據(jù),然后檢查返回結果是否為None,如果不是,則打印找到的數(shù)據(jù)。

0