查找在CSV文件的特定值的單個列，並返回一整行

Question

問題

代碼不正確地識別輸入（項目）。即使CSV文件中存在這樣的值，它也會轉儲到我的失敗消息。任何人都可以幫助我確定我做錯了什麼？

背景

我工作的一個小程序，用戶輸入查詢（這裏沒有給出功能），在搜索CSV文件（項目）的特定列，並返回整行。 CSV數據格式如下所示。我已經縮短了實際數量的數據（49個字段名稱，18000+行）。

代碼

import csv 
from collections import namedtuple 
from contextlib import closing 

def search(): 
    item = 1000001 
    raw_data = 'active_sanitized.csv' 
    failure = 'No matching item could be found with that item code. Please try again.' 
    check = False 

    with closing(open(raw_data, newline='')) as open_data: 
     read_data = csv.DictReader(open_data, delimiter=';') 
     item_data = namedtuple('item_data', read_data.fieldnames) 
     while check == False: 
      for row in map(item_data._make, read_data): 
       if row.Item == item: 
        return row 
       else: 
        return failure  

CSV結構

active_sanitized.csv 
Item;Name;Cost;Qty;Price;Description 
1000001;Name here:1;1001;1;11;Item description here:1 
1000002;Name here:2;1002;2;22;Item description here:2 
1000003;Name here:3;1003;3;33;Item description here:3 
1000004;Name here:4;1004;4;44;Item description here:4 
1000005;Name here:5;1005;5;55;Item description here:5 
1000006;Name here:6;1006;6;66;Item description here:6 
1000007;Name here:7;1007;7;77;Item description here:7 
1000008;Name here:8;1008;8;88;Item description here:8 
1000009;Name here:9;1009;9;99;Item description here:9 

注意

我與Python的經驗是比較少的，但我認爲這將是開始，以便一個很好的問題了解更多。

我確定了打開CSV文件（並封裝在一個關閉函數中）的方法，通過DictReader讀取數據（以獲取字段名稱），然後創建一個命名的元組，以便能夠快速選擇所需的列輸出（項目，成本，價格，名稱）。列順序很重要，因此使用DictReader和namedtuple。

雖然有可能對每個字段名進行硬編碼，但我覺得如果程序可以在文件打開時讀取它們，當處理具有相同列名的類似文件時，它會更有幫助不同的專欄組織。

研究

• CSV頁眉和命名元組： What is the pythonic way to read CSV file data as rows of namedtuples?
• 將CSV數據元組：How to split a CSV row so row[0] is the name and any remaining items are a tuple?
• 有研究額外的鏈接，但我不能發佈超過兩個。

Answer 1

你有三個問題是：

• 將返回第一次失敗，所以它永遠不會讓過去的第一道防線。
• 您正在從文件中讀取字符串，並將其與int進行比較。

• _make遍歷字典鍵而不是值，產生錯誤的結果（item_data(Item='Name', Name='Price', Cost='Qty', Qty='Item', Price='Cost', Description='Description')）。

  for row in (item_data(**data) for data in read_data): 
      if row.Item == str(item): 
       return row 
  return failure 
  

這修復問題在手 - 我們覈對一個字符串，我們只當沒有一個項目中的匹配（儘管你可能要開始轉換字符串到整數的數據，而不是返回這個字符串/整數問題的修復）。

我也改變了你循環的方式 - 使用生成器表達式使得更自然的語法，使用字典命名屬性的常規構造語法。這比使用_make和map()更清潔並且更具可讀性。它還修復了問題3.