检查python字典中是否存在数据

检查python字典中是否存在数据,python,validation,dictionary,key,Python,Validation,Dictionary,Key,我有一个程序,维护一个光盘信息的平面文件数据库。我正在尝试编写一个更新数据库的函数。在这个函数中,我检查艺术家是否存在,如果存在,则将唱片集名称附加到此艺术家,但由于某些原因,它不会看到我键入的艺术家已经存在。我确保键入的内容与字典中的内容完全相同,但由于某种原因,python无法看到它的存在。为什么会发生这种情况?我已经包括了示例输入和python程序。任何帮助都将不胜感激 import sys def add(data, block): artist = block[0]

我有一个程序,维护一个光盘信息的平面文件数据库。我正在尝试编写一个更新数据库的函数。在这个函数中,我检查艺术家是否存在,如果存在,则将唱片集名称附加到此艺术家,但由于某些原因,它不会看到我键入的艺术家已经存在。我确保键入的内容与字典中的内容完全相同,但由于某种原因,python无法看到它的存在。为什么会发生这种情况?我已经包括了示例输入和python程序。任何帮助都将不胜感激

import sys

def add(data, block):
    artist = block[0]
    album = block[1]
    songs = block[2:]
    if artist in data:
        data[artist][album] = songs
    else:
        data[artist] = {album: songs}
    return data


def parseData():

    global data
    file='testdata.txt'
    data = {}
    with open(file) as f:
        block = []
        for line in f:
            line = line.strip()
            if line == '':
                data = add(data, block)
                block = []
            else:
                block.append(line)
        data = add(data, block)

    return data

def artistQry():

    global artists, usrChoiceArt, albums, usrChoiceAlb, usrArtist

    artists=sorted(data.keys())
    for i in range(0,len(artists)) :
        print str(i+1) + " : " + artists[i]



    usrChoiceArt = raw_input("Please choose an artist or enter q to quit:")

    if usrChoiceArt=='q' :
        print "Quitting Now"
        exit()
    else :
        albumQry()

def albumQry():

    global artists, usrChoiceArt, albums, usrChoiceAlb, usrArtist

    usrArtist=artists[int(usrChoiceArt)-1]
    albums=sorted(data[usrArtist].keys())

    for i in range(0,len(albums)) :
        print str(i+1) + " : " + albums[i]

    usrChoiceAlb=raw_input("Please choose an album or enter a to go back:")

    if usrChoiceAlb=="a":
        artistQry()
    else:
        trackQry()


def trackQry():

    global artists, usrChoiceArt, albums, usrChoiceAlb, usrArtist

    usrAlbum=albums[int(usrChoiceAlb)-1]
    tracks=data[usrArtist][usrAlbum]

    for i in range(0,len(tracks)) :
        print tracks[i]

    usrChoiceTrack=raw_input("Enter \"a\" to go back or \"q\" to quit:")

    if usrChoiceAlb=="q":
        print "Quitting Now"
        exit()
    elif usrChoiceTrack=="a":
        albumQry()
    else:
        print "Invalid Choice"
        trackQry()


def artistExist(Name):

    for i in range(0,len(data.keys())):

        if Name==data.keys()[i]:
            return True
        else:
            return False




def updData():

    artistName=raw_input("Please enter an artist name:")
    albumName=raw_input("Please enter an album name:")
    trackList=raw_input("Please enter the track list seperated by comma's:")

    if artistExist(artistName):
        data[artistName].append(albumName)
        print data[artistName]
    elif not artistExist(artistName):
        print "Quitting"
        exit()

if __name__ == '__main__':
data = parseData()


if sys.argv[1]=='-l':
    artistQry()
elif sys.argv[1]=='-a':
    updData()
输入数据:

Bob Dylan
1966 Blonde on Blonde
-Rainy Day Women #12 & 35
-Pledging My Time
-Visions of Johanna
-One of Us Must Know (Sooner or Later)
-I Want You
-Stuck Inside of Mobile with the Memphis Blues Again
-Leopard-Skin Pill-Box Hat
-Just Like a Woman
-Most Likely You Go Your Way (And I'll Go Mine)
-Temporary Like Achilles
-Absolutely Sweet Marie
-4th Time Around
-Obviously 5 Believers
-Sad Eyed Lady of the Lowlands

在函数
artistExist
中,在第一次迭代时返回False!相反,您必须等待所有迭代完成

for i in range(0,len(data.keys())):

    if Name==data.keys()[i]:
        return True
return False

除此之外,这里的
elif
也是多余的:

if artistExist(artistName):
    ...
elif not artistExist(artistName):
    ...
如果事情不是真的,那么它只能是假的。所以你真的应该

if artistExist(artistName):
    ...
else:
    ...
由于函数只是一个不必要的单行线,因此需要一个更好的表达式

if artistName in data:
    ...
else:
    ...

除了通过在循环中返回False而过早返回之外,您做的工作太多了,只需在数据中使用
返回名称

def artistExist(Name):
    return Name in data # will return True or False with O(1) lookup

每次调用
.keys
时,您都在python2中创建一个列表,因此在最坏的情况下,您的查找实际上是二次查找,而不是使用数据中的简单
返回名称
。使用dict的很大一部分是高效的查找,而这会丢失调用键。如果您真的想迭代这些键,您只需对数据中的键执行
,不需要调用.keys,也不需要使用range。

是的,也是这样。我有点被OP为什么循环弄糊涂了。@chrisaycock,是的,我会改变很多事情,我只是选择了这个,因为它看起来很痛苦,因为它效率太低了!这也是你的吗?我注意到你们都有一个输入完全相同的问题。