루프 내 중첩 된 사전

Question

다음과 같은 모양의 CSV 파일이 있습니다. 이것은 1980 년부터 2014 년까지 4 개국의 전기 소비를 보여줍니다. 중첩 된 사전을 만들려고합니다. 예를 들어 consumption['United States'][1980]이 올바른 값을 반환합니다. . 내가 정수 년의 목록과 배열을 가지고 있고이 같은 사전을 만들려고 해요 :

값은 해당 연도의 값을 포함하는 배열입니다

file = open('power dataset.csv', 'r') 

years = list(range(1980, 2015)) 

consumption = {} 
generation = {} 

generation = False 

for line in file: 

    if("Nuclear" in line): 
     break 

    split = line.split(",") 

    if split[0] == "Generation": 
     generation = True 

    if "Egypt" == split[0] or split[0] == "Germany" or split[0] == "Netherlands" or split[0] == "United States": 

     values = split[2:] 

     if not generation: 

      i = 0 

      for year in years: 
       country = split[0] 
       consumption[country] = {year: values[i]} 
       i = i+1 

. 내가 겪고있는 문제는 사전이 오직 1 년과 하나의 값 (마지막 것)을 포함한다는 것입니다. 따라서 consumption['United States'][1980]과 같은 것을 인쇄하려고하면 사전에 1980 년에 항목이 없기 때문에 2014 년에만 오류가 발생합니다.

나는 아주 간단한 것을 놓치고있는 것처럼 느껴지지만 그 위에는 내 손가락을 대지 못합니다. .

Here은 전체 CSV 파일입니다.

for year in years: 
    consumption[country] = {year: values[i]} 

당신은 루프의 각 반복에 consumption[country]의 이전 값을 덮어 쓰기 :

Answer 1

문제는이와 그 것으로 보인다.

대신

,이 시도 : 사전 - 이해의

if country in ("Egypt", "Germany", "Netherlands", "United States"): 
    if not generation: 
     consumption[country] = {year: vals for year, vals in zip(years, values)} 

단계별 고장 예를 zip로 :

>>> years = [1980, 1981, 1982, 1983] 
>>> values = [1, 2, 3, 4] 
>>> zip(years, values) 
[(1980, 1), (1981, 2), (1982, 3), (1983, 4)] 
>>> {year: vals for year, vals in zip(years, values)} 
{1980: 1, 1981: 2, 1982: 3, 1983: 4} 

을 다른 방법으로, 내부 루프 전에 consumption[country] = {}로 consumption[country]를 초기화 할 수 편집하기 전에 원래 코드에서와 같이 consumption[country][year] = values[i]을 사용하십시오.