uaveiro-leci/1ano/fp/aula05/tp05-codecheck/exercises02/ex01.py

32 lines
938 B
Python

# Substitua as reticências pelo valor adequado para obter as respostas pedidas.
# Considere que num comboio de passageiros
# as primeiras N carruagens têm lugares de primeira classe
# enquanto as restantes têm lugares de segunda classe.
# Complete a definição para que passengers(train, n)
# devolva uma lista com dois elementos, [num1, num2], onde:
# num1 = número de passageiros em primeira classe (nas n primeiras carruagens)
# num2 = número de passageiros em segunda classe.
def passengers(train, n):
class1 = train[:n]
class2 = train[n:]
num1 = sum(class1)
num2 = sum(class2)
return [num1, num2]
def test():
train1 = [12, 32, 10, 21]
train2 = [9, 29, 19]
train3 = [14, 34, 24]
print(train1, train2, train3)
print('\nTesting passengers')
print(passengers(train1, 2))
print(passengers(train2, 1))
print(passengers(train3, 0))
# Run tests:
test()
# JMR@ua.pt 2021