要查找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
,如果不是,則打印找到的數據。