def countPrefix(words, prefix):
return len([1 for w in words if w.startswith(prefix)])
words = ['rato', 'roeu', 'rolha', 'rainha', 'rei', 'russia']
pref = ['ro', 'ra', 'r']
result = [countPrefix(words, p) for p in pref]
print result
ZGVmIGNvdW50UHJlZml4KHdvcmRzLCBwcmVmaXgpOgogICAgcmV0dXJuIGxlbihbMSBmb3IgdyBpbiB3b3JkcyBpZiB3LnN0YXJ0c3dpdGgocHJlZml4KV0pCgp3b3JkcyA9IFsncmF0bycsICdyb2V1JywgJ3JvbGhhJywgJ3JhaW5oYScsICdyZWknLCAncnVzc2lhJ10KcHJlZiA9IFsncm8nLCAncmEnLCAnciddCgpyZXN1bHQgPSBbY291bnRQcmVmaXgod29yZHMsIHApIGZvciBwIGluIHByZWZdIAoKcHJpbnQgcmVzdWx0