在python中用if/else語句在CSV查找中使用通配符

Question

這段代碼來自一個python腳本，我將在CSV文件中搜索員工的員工id列，然後輸出它，如果它存在於柱。

我想要做的是，如果有人更改CSV文件中的列順序，通過檢查列的標題，在csv文件中向我拋出列3不再是員工ID列的錯誤，通過使用通配符來查看它是否包含員工ID的一部分。

import csv 

with open('Report1.csv', 'rb') as user_file: 
    reader = csv.reader(user_file) 

    for column in reader: 
     print column[2] 
     employeeIid = column[2] 

     if column[2] == 'Employee ID': 
      print employeeIid 
     else: 
      print "Employee ID Column in CSV changed-please repair" 

當我運行上面的代碼中，我得到這樣的：

Employee ID 
123456 
Employee ID Column in CSV changed-please repair 

它打印員工ID，以及警告錯誤。我如何改進/修復我的代碼？

請注意：員工ID是CSV文件中的第3列（不是2-位，我認爲Python使用0作爲開始）。

另外，我該如何忽略標題，並且只打印員工ID號（例如123456）而不是頭和實際的號碼？

Answer 1

看起來你的代碼混淆了行和列。

reader的每次迭代都會返回一行數據。 第一次迭代從您的csv文件返回header/columns。

另外，使用r而不是rb以文本模式打開文件。

例子。

import csv 

with open('Report1.csv', 'r') as user_file: 
    reader = csv.reader(user_file) 

    for row_idx, row in enumerate(reader): 
     # check if header row 
     if row_idx == 0: 
      # check column 2 of this row 
      if row[2].strip() != "Employee ID": 
       print("Employee ID Column in CSV changed-please repair") 
       break 
     else: 
      # to only print "Employee ID" 
      print(row[2]) 
      # to print all columns in the row 
      # print(', '.join(row))