Python словари с файлами

Question

у меня есть .txt файл, содержащий строки текста, как это:

Godbals NO LO -1 -1 1 1 1 -1 -1 1 1 1 1 1 1 1 -1 1 1 1 -1 1 1 1 1 1 -1 1 -1 -1 1 1 1 1 1 1 0 0 1 -1 -1 1 -1 1 -1 1 1 -1

Моя задача для определения процедуры, которая возвращает словарь, ключи которого являются первым элементом строк текста, то есть «Godbals», а значения - это числа -1,1,0.

я импортировал этот файл с помощью открытого метода и назначен его переменной strlist

Это часть кода я написал;

def create_voting_dict(strlist): 
    D = {} 
    splitlist = [k.split() for k in strlist] 
    for i in range(len(strlist)): 
     intlist = [int(j) for j in splitlist[i][3:]] 
     D.update({splitlist[i][0]:None }) 
     for p in intlist: 
      D[splitlist[i][0]] = p 
    return D 

Это возвращает этот словарь;

{'Cornyn': 1, 'McConnell': 1, 'Bingaman': -1, 'Talent': 1, 'Cochran': 1, 'Craig': -1, 'Lugar': 1, 'Clinton': 1, 'Domenici': 1, 'Rockefeller': -1, 'Gregg': 1, 'Lautenberg': -1, 'Chambliss': 1, 'Voinovich': -1, 'Murkowski': 1, 'Levin': -1, 'Pryor': 1, 'Martinez': 1, 'Harkin': -1, 'Johnson': 1, 'Bayh': -1, 'Byrd': -1, 'Coburn': 1, 'Bunning': 1, 'Reid': -1, 'Kennedy': -1, 'Shelby': 0, 'Sarbanes': -1, 'Bond': 1, 'Nelson1': 1, 'Dole': 1, 'Coleman': 1, 'Dorgan': -1, 'Durbin': -1, 'Reed': -1, 'Landrieu': 1, 'Schumer': 1, 'Santorum': 1, 'Snowe': 1, 'Lieberman': -1, 'Hutchison': 1, 'Lincoln': 1, 'DeMint': 1, 'Isakson': 1, 'Inhofe': 1, 'Dodd': -1, 'DeWine': 1, 'Frist': 1, 'Bennett': 1, 'Smith': 1, 'Thomas': 1, 'Enzi': 1, 'Thune': 1, 'Burr': -1, 'Boxer': -1, 'Allen': 1, 'Nelson2': 1, 'Kerry': -1, 'Murray': -1, 'Burns': 1, 'Allard': 1, 'Salazar': 1, 'Jeffords': -1, 'Akaka': -1, 'Sununu': 1, 'Biden': -1, 'Obama': -1, 'Cantwell': 1, 'Roberts': 1, 'Alexander': 1, 'McCain': 1, 'Collins': 1, 'Inouye': -1, 'Lott': 0, 'Feingold': -1, 'Kyl': 1, 'Feinstein': 1, 'Kohl': -1, 'Specter': 1, 'Stevens': 1, 'Ensign': 1, 'Carper': 1, 'Hatch': 1, 'Vitter': 1, 'Baucus': 1, 'Graham': 1, 'Grassley': 1, 'Wyden': -1, 'Hagel': 1, 'Dayton': 1, 'Mikulski': -1, 'Stabenow': 1, 'Leahy': -1, 'Sessions': 1, 'Crapo': 1, 'Conrad': -1, 'Warner': 1, 'Chafee': -1, 'Brownback': 1} 

Я вижу, как мой код несовершенен, я хотел бы найти способ итеративно добавляемый к -1,0,1, чтобы значению элементов словаря ... или что-то в этом роде.

Answer 1

Вы можете split вашу строку с maxsplit 3, и использовать set:

>>> s="Godbals NO LO -1 -1 1 1 1 -1 -1 1 1 1 1 1 1 1 -1 1 1 1 -1 1 1 1 1 1 -1 1 -1 -1 1 1 1 1 1 1 0 0 1 -1 -1 1 -1 1 -1 1 1 -1" 
>>> l=s.split(' ',3) 
>>> d={l[0]:set(l[-1].split())} 
>>> d 
{'Godbals': set(['1', '0', '-1'])} 

Answer 2

Вы можете начать разделив строку пробелом

split_line = line.split(" ") 

И затем, начать индексацию на него. Насколько я понимаю, ваш ключ будет,

key = split_line[0] 
value = split_line[3:] 
final_dict = {key : value} 

Надежда, что помогает