from p2finished import * rooms = [200, 400, 200, 100] capacities = [rabbit_house(rs, 10, "normal") for rs in rooms] print(capacities) print(sum(capacities)) rooms = [] # Kitchen rooms.append([200, 10, "light"]) # Living Room rooms.append([400, 13, "normal"]) # Bedroom rooms.append([200, 3, "light"]) # Bathroom rooms.append([100, 16, "light"]) print(rooms) # rhp : rabbit house parameters capacities = [rabbit_house(*rhp) for rhp in rooms] print(capacities)