私は、名前のリストを取得し、性別に関係なく、上位 20 件の名前を返すプログラムに取り組んでいます。
insert_sort を使用しようとしていて、機能していないため、自分が間違っていることを理解するのに少し苦労しています。
これがコードです。
class babyName():
__slots__ = ("name" , "gender" , "occurrences")
def __init__(self, name, gender, occurrences):
self.name = name
self.gender = gender
self.occurrences = occurrences
def __str__(self):
return self.name + "," + self.gender + "," + self.occurrences
def mkEntry(name, gender, occurrences):
entry = babyName(name, gender, occurrences)
return entry
def readFile(year):
lst = []
lst2 = []
topm = 20
topf = 20
yob = ("yob" + str(year) + ".txt")
for line in open("names/" + yob ):
line = line.split(",")
if line[1] == "F":
if topf > 0:
entry = mkEntry(line[0], line[1], line[2])
lst.append(entry)
topf -= 1
elif line[1] == "M":
if topm > 0:
entry = mkEntry(line[0], line[1], line[2])
lst2.append(entry)
topm -= 1
mergedlist = lst + lst2
return mergedlist
def insertion_sort(lst):
for i in range(1, len(lst)):
h = i - 1
val = lst[i].occurrences
val2 = lst[i-1].occurrences
while (h >= 0) and (val2 > val):
lst[h+1] = lst[h]
h = h - 1
lst[h+1] = val
def main():
year = input("Enter the year of birth:")
file = readFile(year)
insertion_sort(file)
for x in file:
print(x)
main()
なにが問題ですか？