Comment obtenir le nombre total de lignes d'un fichier CSV en Python ?

Question

J'utilise python (Django Framework) pour lire un fichier CSV. Je ne tire que 2 lignes de ce CSV comme vous pouvez le voir. Ce que j'ai essayé de faire est de stocker dans une variable le nombre total de lignes du CSV.

Comment puis-je obtenir le nombre total de lignes ?

file = object.myfilePath
fileObject = csv.reader(file)
for i in range(2):
    data.append(fileObject.next()) 

J'ai essayé :

len(fileObject)
fileObject.length

Answer 1

Si vous devez analyser le CSV (par exemple, en raison de la présence de sauts de ligne dans les champs ou de lignes commentées) mais que le CSV est trop volumineux pour occuper la mémoire en une seule fois, vous pouvez analyser le CSV morceau par morceau :

import pandas as pd
import os
import sys

csv.field_size_limit(sys.maxsize)  # increase the maximal line length in pd.read_csv()

cnt = 0
for chunk in pd.read_csv(filepath, chunksize=10**6):
    cnt += len(chunk)
print(cnt)

Answer 2

Je pense que la mienne sera l'approche la plus simple ici :

import csv
file = open(filename, 'r')
csvfile = csv.reader(file)
file.close
print("row", len(list(csvfile)))

Answer 3

Si vous travaillez sur un système Unix, la méthode la plus rapide est la commande shell suivante

cat FILE_NAME.CSV | wc -l

Depuis Jupyter Notebook ou iPython, vous pouvez l'utiliser avec une ! :

! cat FILE_NAME.CSV | wc -l

Answer 4

Essayez

data = pd.read_csv("data.csv")
data.shape

et dans la sortie vous pouvez voir quelque chose comme (aa,bb) où aa est le nombre de lignes.