Puis-je écrire ceci comme une compréhension de liste à deux niveaux?

Question

J'ai ce bit:

for line in entireResult.split('\n'): 
    print line.split(',')[0] 

... et je me sens comme il peut être liste-compris, mais je suis un peu trop vendredi-nighted. J'ai essayé quelque chose comme:

[for l[0] in line.split(',') for line in entireResult('\n')] 

mais cela n'a pas volé (ligne non définie). Y a-t-il un moyen de faire cela? Super karma supplémentaire pour répondre avant de finir de coder mon script avec le codez de la boucle ennuyeuse.

Answer 1

Vous avez seulement une boucle.

[line.split(',')[0] for line in entireResult.split('\n')] 

Bien sûr, vous devriez le faire en tant que générateur si possible.

print '\n'.join(line.split(',')[0] for line in entireResult.split('\n'))