Pythonのif/else文でのワイルドカードの使用

Question

このコードは、私が持っているPythonスクリプトのものです。従業員のCSVファイル内の従業員ID列を検索して出力しますカラム。

誰かがCSVファイルの列の順序を変更した場合、csvファイルの列のヘッダーを確認しても列3がEmployee ID列ではないというエラーが表示されるワイルドカードを使用してEmployee 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は1行のデータを返します。 最初の反復では、CSVファイルからheader/columnsが返されます。

さらに、rbの代わりにrを使用してファイルをテキストモードで開きます。

例。

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))