Файл: Найдите количество кубиков каждого из заданных цветов и их суммарный объём

from collections import defaultdict
 
def getDataFromFile(path='in.txt'):
    d=defaultdict(list)
    with open(path,'r') as file:
        [d[line.split()[1]].append((line.split()[0],line.split()[2])) for line in file.read().split('\n')]
    return d    
 
di=getDataFromFile()
print('Количество кубиков каждого из перечисленных цветов')
for key in di.keys():
    print(key,": ",len(di[key]),end="\n")    
s=0
d3=0
m5=0
values=di.values()
print(values)
for val in values:
    for v in val:
        s+=pow(int(v[0]),3)
        if v[1]=='деревянный' and int(v[0])==3:
            d3+=1
        if v[1]=='металлический' and int(v[0])>5:
            m5+=1
            
print('Cуммарный объём: ',s,end='\n')     
print('Количество деревянных кубиков с ребром 3 см ',d3)            
print('Количество металлическиx кубиков с ребром > 5 см ',m5)

from collections import defaultdict
 
def getDataFromFile(path='in.txt'):
    d=defaultdict(list)
    with open(path,'r') as file:
        [d[line.split()[1]].append((line.split()[0],line.split()[2])) for line in file.read().split('\n')]
    return d    
 
di=getDataFromFile()
print('Количество кубиков каждого из перечисленных цветов')
for key in di.keys():
    print(key,": ",len(di[key]),end="\n")    
values=di.values()
print('Cуммарный объём: ',sum([sum([pow(int(v[0]),3) for v in val])for val in values]),end='\n')
print('Количество деревянных кубиков с ребром 3 см ',sum([sum([1 for x in val if x[1]=='деревянный' and int(x[0])==3])for val in values]),end='\n')
print('Количество металлическиx кубиков с ребром > 5 см ',sum([sum([1 for x in val if x[1]=='металлический' and int(x[0])>5])for val in values]),end='\n')