检查python字典中是否存在数据
我有一个程序,维护一个光盘信息的平面文件数据库。我正在尝试编写一个更新数据库的函数。在这个函数中,我检查艺术家是否存在,如果存在,则将唱片集名称附加到此艺术家,但由于某些原因,它不会看到我键入的艺术家已经存在。我确保键入的内容与字典中的内容完全相同,但由于某种原因,python无法看到它的存在。为什么会发生这种情况?我已经包括了示例输入和python程序。任何帮助都将不胜感激检查python字典中是否存在数据,python,validation,dictionary,key,Python,Validation,Dictionary,Key,我有一个程序,维护一个光盘信息的平面文件数据库。我正在尝试编写一个更新数据库的函数。在这个函数中,我检查艺术家是否存在,如果存在,则将唱片集名称附加到此艺术家,但由于某些原因,它不会看到我键入的艺术家已经存在。我确保键入的内容与字典中的内容完全相同,但由于某种原因,python无法看到它的存在。为什么会发生这种情况?我已经包括了示例输入和python程序。任何帮助都将不胜感激 import sys def add(data, block): artist = block[0]
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,是的,我会改变很多事情,我只是选择了这个,因为它看起来很痛苦,因为它效率太低了!这也是你的吗?我注意到你们都有一个输入完全相同的问题。