Python CSV dictdicter，未获得预期的输出

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

预期输出：

实际输出：

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遍历第一行（约翰·史密斯）时，您将强制line_count2 == 0何时打印标题。

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

您可以简化功能：

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