def es_nom_de_noi(s):
return s in ['Albert', 'Antoni', 'Lluis', 'Marc',
'Pere', 'Pol', 'Ramon']
def primers_noms(f, n):
'''
>>> primers_noms(['Ramon', 'Laia', 'Gina', 'Alba', 'Aina', 'Lluis'], 2)
['Ramon', 'Lluis']
>>> primers_noms(['Ramon', 'Laia', 'Gina', 'Alba', 'Aina', 'Lluis'], 3)
['Ramon', 'Lluis', 'i cap mes']
>>> primers_noms(['Laia', 'Gina', 'Alba', 'Aina'], 2)
['i cap mes']
>>> primers_noms(['Pere'], 0)
[]
>>> primers_noms([], 5)
['i cap mes']
'''
llista = []
c = 0
if n == 0:
return llista
for i in f:
if es_nom_de_noi(i):
llista += [i]
c += 1
if c == n:
return llista
if c < n:
llista += ['i cap mes']
return llista
if __name__ == "__main__":
import doctest
doctest.testmod(verbose=True)