питона - Подняв ключ словаря в другом файл с двумя критериями

Question

После окончания моего кода, у меня есть словарь, как так:

{'"WS1"': 1475.9778073075058, '"BRO"': 1554.1437268304624, '"CHA"': 1552.228925324831} 

То, что я хочу сделать, это найти каждый из ключей в отдельный файл, teams.txt, который отформатирован так:

1901,'BRO','LAD' 
1901,'CHA','CHW' 
1901,'WS1','MIN' 

Использование года, что 1901, и команда, которая является ключом каждого элемента в словаре, я хочу, чтобы создать новый словарь, где ключ является третьим столбцом в файле teams.txt, если год и команда совпадают, а значение - это значение команды в первый словарь.

Я полагал, что это было бы проще всего, если бы я создал функцию «поиска» года и команды и вернул «franch», а затем применил эту функцию к каждому ключу в словаре. Это то, что я до сих пор, но это дает мне KeyError

def franch(year, team_str): 
    team_str = str(team_str) 
    with open('teams.txt') as imp_file: 
    teams = imp_file.readlines() 
    for team in teams: 
    (yearID, teamID, franchID) = team.split(',') 
    yearID = int(yearID) 
    if yearID == year: 
     if teamID == team_str: 
     break 
    franchID = franchID[1:4] 
    return franchID 

А в другой функции со словарем, что я хочу, чтобы применить эту функцию:

franch_teams={} 
    for team in teams: 
    team = team.replace('"', "'") 
    franch_teams[franch(year, team)] = teams[team] 

Идеальный выход, что Я пытаюсь выполнить, будет выглядеть так:

{'"MIN"': 1475.9778073075058, '"LAD"': 1554.1437268304624, '"CHW"': 1552.228925324831} 

Спасибо!

Answer 1

Имеет ли этот набор кодов ваши потребности?

Я делаю дополнительную проверку на равенство, потому что в разных частях вашего кода были разные строковые знаки.

def almost_equals(one, two): 
    one = one.replace('"', '').replace("'", "") 
    two = two.replace('"', '').replace("'", "") 
    return one == two 

def create_data(year, data, text_content): 
    """ This function returns new dictionary. """ 
    content = [line.split(',') for line in text_content.split('\n')] 
    res = {} 

    for key in data.keys(): 
     for one_list in content: 
      if year == one_list[0] and almost_equals(key, one_list[1]): 
       res[one_list[2]] = data[key] 

    return res 


teams_txt = """1901,'BRO','LAD' 
1901,'CHA','CHW' 
1901,'WS1','MIN'""" 

year = '1901' 
data = { '"WS1"': 1475.9778073075058, '"BRO"': 1554.1437268304624, '"CHA"': 1552.228925324831 } 
result = create_data(year, data, teams_txt) 

И выход:

{"'CHW'": 1552.228925324831, "'LAD'": 1554.1437268304624, "'MIN'": 1475.9778073075058} 

Update:

Для чтения из текстового файла используйте эту функцию:

def read_text_file(filename): 
    with open(filename) as file_object: 
     result = file_object.read() 

    return result 

teams_txt = read_text_file('teams.txt') 

Answer 2

Вместо того, чтобы нарушать цикл, пройдите его и смените ключ для franchID.
Учитывая dic ваш базовый словарь (в вашем примере, {'"WS1"': 1475.9778073075058, '"BRO"': 1554.1437268304624, '"CHA"': 1552.228925324831}) и с помощью кода, вы можете сделать:

def franch(year,team_str): 
    team_str=str(team_str) 
    with open('teams.txt') as imp_file: 
     teams=imp_file.readlines() 
    for team in teams: 
     (yearID,teamID,franchID)=team.split(',') 
     yearID=int(yearID) 
     if yearID==year: 
      if teamID==team_str: 
       #Found 
       dic[franchID] = dic[teamID] 
       del dic[teamID] 

Эта часть кода

dic[franchID] = dic[teamID] 
del dic[teamID] 

Будет ли удалить «изменить» teamID для franchID, создав новую запись с идентификатором franchID и удалив старую запись, поэтому выход будет

{'"MIN"': 1475.9778073075058, '"LAD"': 1554.1437268304624, '"CHW"': 1552.228925324831} 

, как ожидалось.

Если вы хотите сохранить старый словарь, вам не нужно удалять ключ. Просто сделайте

dic2[franchID] = dic1[teamID] 

и вы получите два словаря. Просто объявите dic2 = dict(), чтобы не получить NameError.

Answer 3

Вы можете попробовать что-то вроде:

#!/usr/bin/env python 

def clean(_str): 
    return _str.strip('"').strip("'") 

first = {'"WS1"': 1475.9778073075058, '"BRO"': 1554.1437268304624, '"CHA"': 1552.228925324831} 

clean_first = dict() 
second = dict() 

for k,v in first.items(): 
    clean_first[clean(k)] = v 

with open("teams.txt", "r") as _file: 
    lines = _file.readlines() 
    for line in lines: 
     _,old,new = line.split(",") 
     second[new.strip()] = clean_first[clean(old)] 

print second 

Что дает ожидаемый:

{"'CHW'": 1552.228925324831, "'LAD'": 1554.1437268304624, "'MIN'": 1475.9778073075058}