Imprimer les deux premières lignes d'un fichier csv pour une sortie standard

Je voudrais imprimer (stdout) les deux premières lignes d'un fichier csv:

#!/usr/bin/env python
import csv
afile = open('<directory>/*.csv', 'r+')
csvReader1 = csv.reader(afile)
for row in csvReader1:
    print row[0]
    print row[1]

cependant, mon résultat à l'aide de ce code d'imprimer les deux premières colonnes.

Des suggestions?

OriginalL'auteur Czed | 2011-10-05

  1. 14

    Vous souhaitez imprimer une ligne, mais votre code demande d'impression de la première et deuxième membres de chaque ligne

    Puisque vous voulez imprimer l'ensemble de la ligne, vous pouvez simplement l'imprimer, et en outre, lire les deux premiers

    #!/usr/bin/env python
import csv
afile = open('<directory>/*.csv', 'r+')
csvReader1 = csv.reader(afile)
for i in range(2):
    print csvReader1.next()
    Bon montage, mais pas avant compatible -- .next() est .__next__() sur Python 3.
    +1 pour la structure en boucle, ce qui permet pour l'impression de la première n lignes du fichier
    Merci! Cela semble d'imprimer les deux premières lignes après les informations d'en-tête. J'espérais pour imprimer l'en-tête de ligne, et ensuite la ligne suivante en dessous.
    Utilisation 'rb' pour ouvrir des fichiers csv pour en Python 2.X (voir la section csv.lecteur).
    Suggérer next(csvReader1) (fonctionne dans les deux Python 2.X et 3.X).

    OriginalL'auteur Ofir

  2. 5

    Généralement, vous pouvez limiter les itérateurs avec le itertools.islice fonction:

    import itertools
for row in itertools.islice(csvReader1, 2):
    print row

    Ou par une utilisation créative de zip():

    for line_number, row in zip(range(2), csvReader1):
    print row

    OriginalL'auteur Petr Viktorin