Python CSV dictReader，没有得到预期 output

Question

下面是代码，打开文件并将其传递给 python 中的 CSV 处理程序

def csv_reader_dict():
  #Reading Data From CSV file into Dictionary using csv
  print("---------Reading Data From CSV file into Dictionary using csv------------")
  with open('input.txt','r') as file_data:
    csv_format = csv.DictReader(file_data)
    line_count2 = 0
    for row in csv_format:
        if line_count2 == 0:
            print("Header are {}".format(",".join(row) ))
            line_count2 += 1
        else:
            print("{} reside at {} and joined on {}".format(row["name"],row["address"],row["date joined"]))
            line_count2 += 1
    print("Processed {} line count".format(line_count2))
  print("------------------------------------------------------------------------")
  print("\n")

csv_reader_dict()

预期 Output：

实际 Output：

Answer 1

DictReader 假定第一行是标题，因此它会跳过它。 在您的代码中，您正在使用第一次迭代来打印标题而不是所需的信息。

将fieldnames名添加到DictReader应该可以工作，因为它不会跳过第一行，并且您的代码将按预期工作。

def csv_reader_dict():
#Reading Data From CSV file into Dictionary using csv
print("---------Reading Data From CSV file into Dictionary using csv------------")
with open('input.txt','r') as file_data:
    csv_format = csv.DictReader(file_data, fieldnames=['name', 'address', 'date joined'])
    line_count2 = 0
    for row in csv_format:
        if line_count2 == 0:
            print("Header are {}".format(",".join(row) ))
            line_count2 += 1
        else:
            print("{} reside at {} and joined on {}".format(row["name"],row["address"],row["date joined"]))
            line_count2 += 1
    print("Processed {} line count".format(line_count2))
print("------------------------------------------------------------------------")
print("\n")

Answer 2

当line_count2 == 0时，它会遍历第一行（john smith），而当line_count2 == 0时，您会强制打印 header。

您认为行print("Header are {}".format(",".join(row) ))它正在打印 header，但它正在打印第一行的键，它们是相同的标题。

您可以简化 function：

def csv_reader_dict():
    #Reading Data From CSV file into Dictionary using csv
    print("---------Reading Data From CSV file into Dictionary using csv------------")
    with open('input.txt','r') as file_data:
        csv_format = csv.DictReader(file_data)
        line_count2 = 0
        print("Header are {}".format(",".join(csv_format.fieldnames)))
        for row in csv_format:
            print("{} reside at {} and joined on {}".format(row["name"],row["address"],row["date joined"]))
            line_count2 += 1
    print("Processed {} line count".format(line_count2))
    print("------------------------------------------------------------------------")
    print("\n")

csv_reader_dict()