带有 csv 文件的字典 python

Question

我是所有这些编码方面的新手，但即使事情不起作用，我也有很多动力。

我正在尝试使用 CSV 文件处理字典。我可以打开并编辑一个新文件，但我想我在尝试从 csv 文件中读取时遗漏了一些东西。 我在第 39-41 行遇到错误，我可能做错了什么，但它是什么？ 代码如下：

import csv
import os
import os.path

phonebook = {}
def print_menu():
    print('1. Print phone book')
    print('2. add phone book')
    print('3. look for number using first name only')
    print('4. look by last name')
    print('5. exit')
    print()

menu_selection = 0

while menu_selection != 5:

    if os.path.isfile('my_phonebook.csv') == True:  
        csv_file = open('my_phonebook.csv', 'a', newline = '')
        writer = csv.writer(csv_file)
    else:
        csv_file = open('my_phonebook.csv', 'w', newline = '')
        writer = csv.writer(csv_file)
        headers = ['first_name','last_name','phone_number']
        writer.writerow(headers)
    print_menu()
    menu_selection = int(input('type menu selection number'))

    if menu_selection == 2:  #add list in phone book
        first_name = input("enter first name: ")
        last_name = input("enter last name: ")
        phone_number = input("enter phone number: ")
        writer.writerow([first_name,last_name,phone_number])
        csv_file.close()
    elif menu_selection == 1: #print phone book
        print("phone book:")
        listGen = csv.reader(csv_file, delimiter=' ', quotechar='|') #error starts here and in the next two rows...
        for row in csv_file:
            print(row)
    elif menu_selection == 3: #look for number using first name only
        print('look up number')
        first_name = input('first name:')
        if first_name in phonebook:
            print('the number is', phonebook[phone_number])
        else:
            print('name not found')
    elif menu_selection == 4: #print all details of last name entered
        print('search by last name')
        last_name = input('please enter last name: ')
        for row in csv_file:
            print(row)

Answer 1

文件有内容吗？看起来您正在尝试遍历一个空的行迭代器。尝试类似...

for row in csv_file:
    print(row)
else:
    print('no phone numbers')

看看你会得到什么。

Answer 2

请检查以下代码。 我还尝试修复了一些其他错误。针对相同的问题进行了一些更改。

为了阅读 csv，我没有使用 csv 阅读器模块。

如果你想使用它，请替换 search() 函数中的代码。

import csv
import os
import os.path

phonebook = {}
def print_menu():
    print('1. Print phone book')
    print('2. add phone book')
    print('3. look for number using first name only')
    print('4. look by last name')
    print('5. exit')
    print()

def search(name):
    phonebook = open('my_phonebook.csv','r')
    name_found = True
    for line in phonebook:
        if name in line:
            fname,lname,num = line.split(',')
            print "First Name is ",fname.strip()
            print "Last Name is ",lname.strip()
            print 'the number is', num.strip()

            name_found = True
    if  not name_found:
        print('name not found')
    return    
menu_selection = 0

while menu_selection != 5:


    print_menu()
    menu_selection = int(input('type menu selection number - '))

    if menu_selection == 2:  #add list in phone book
        if os.path.isfile('my_phonebook.csv') == True:  
            csv_file = open('my_phonebook.csv', 'a')
            writer = csv.writer(csv_file)
        else:
            csv_file = open('my_phonebook.csv', 'w')
            writer = csv.writer(csv_file)
            headers = ['first_name','last_name','phone_number']
            writer.writerow(headers)
        first_name = input("enter first name: ")
        last_name = input("enter last name: ")
        phone_number = input("enter phone number: ")
        writer.writerow([first_name,last_name,phone_number])
        csv_file.close()

    elif menu_selection == 1: #print phone book
        print("phone book:")
        if os.path.isfile('my_phonebook.csv') == True:  
            csv_file=open('my_phonebook.csv','r')

            for row in csv_file:
                print(row)
            csv_file.close()
        else:
            print "Phone book file not created. First create it to read it"

    elif menu_selection == 3: #look for number using first name only
        print('look up number')
        first_name = input('first name:')
        search(first_name)


    elif menu_selection == 4: #print all details of last name entered
        print('search by last name')
        last_name = input('please enter last name: ')
        search(last_name)

输出：

C:\Users\dinesh_pundkar\Desktop>python b.py
1. Print phone book
2. add phone book
3. look for number using first name only
4. look by last name
5. exit
()
type menu selection number - 1
phone book:
Phone book file not created. First create it to read it
1. Print phone book
2. add phone book
3. look for number using first name only
4. look by last name
5. exit
()
type menu selection number - 2
enter first name: "Dinesh"
enter last name: "Pundkar"
enter phone number: "12345"
1. Print phone book
2. add phone book
3. look for number using first name only
4. look by last name
5. exit
()
type menu selection number - 1
phone book:
first_name,last_name,phone_number

Dinesh,Pundkar,12345

1. Print phone book
2. add phone book
3. look for number using first name only
4. look by last name
5. exit
()
type menu selection number - 3
look up number
first name:"Dinesh"
First Name is  Dinesh
Last Name is  Pundkar
the number is 12345
1. Print phone book
2. add phone book
3. look for number using first name only
4. look by last name
5. exit
()
type menu selection number - 4
search by last name
please enter last name: "Pundkar"
First Name is  Dinesh
Last Name is  Pundkar
the number is 12345
1. Print phone book
2. add phone book
3. look for number using first name only
4. look by last name
5. exit
()
type menu selection number - 5

C:\Users\dinesh_pundkar\Desktop>