Converted names.py to a list comprehension
This commit is contained in:
parent
e8319244b3
commit
1c2f80684e
|
@ -1,12 +1,9 @@
|
||||||
def main():
|
def main():
|
||||||
with open('names.txt', 'r') as f:
|
with open('names.txt', 'r') as f:
|
||||||
names = f.read().split('\n')
|
names = f.read().split('\n')
|
||||||
surnames = set([name.split(' ')[-1] for name in names[1:]])
|
namesDict = {surname: set([name.split(' ')[0] for name in names[1:] if surname in name])
|
||||||
namesDict = {}
|
for surname in set([name.split(' ')[-1] for name in names[1:] if name != ''])}
|
||||||
for surname in surnames:
|
|
||||||
namesDict[surname] = set([name.split(' ')[0] for name in names[1:] if surname in name and surname != ''])
|
|
||||||
|
|
||||||
del namesDict['']
|
|
||||||
printNames(namesDict)
|
printNames(namesDict)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue