Hvordan lese CSV-fil i Python (Eksempler på modul, pandaer)

CSV er et vanlig format for datautveksling da det er kompakt, enkelt og generelt. Mange nettjenester lar brukerne eksportere tabelldata fra nettstedet til en CSV-fil. CSV-filer åpnes i Excel, og nesten alle databaser har et verktøy for å tillate import fra CSV-filer.

CSV-eksempelfil

Data i form av tabeller kalles også CSV (kommaseparerte verdier) - bokstavelig talt "kommaseparerte verdier." Dette er et tekstformat beregnet for presentasjon av tabelldata. Hver linje i filen er én linje i tabellen. Verdiene til individuelle kolonner er atskilt med et skillesymbol – et komma (,), et semikolon (;) eller et annet symbol. CSV kan enkelt leses og behandles av Python.

Tenk på følgende tabell

Tabelldata

Du kan representere denne tabellen i csv som nedenfor.

CSV-data

Programmeringsspråk, designet av, dukket opp, utvidelse

Python, Guido van Rossum, 1991, .py

Java, James Gosling, 1995, .java

C++, Bjarne Stroustrup,1983,.cpp

Som du kan se er hver rad en ny linje, og hver kolonne er atskilt med komma. Dette er et eksempel på hvordan en CSV-fil ser ut.

Last ned CSV-data

Python CSV-modul

Python tilbyr en CSV-modul for å håndtere CSV-filer. For å lese/skrive data må du gå gjennom radene i CSV-filen. Du må bruke splittmetoden for å hente data fra spesifiserte kolonner.

CSV-modulfunksjoner

I CSV-moduldokumentasjonen kan du finne følgende funksjoner:

• csv.field_size_limit – returner maksimal feltstørrelse
• csv.get_dialect – få dialekten som er knyttet til navnet
• csv.list_dialects – vis alle registrerte dialekter
• csv.reader – les data fra en csv-fil
• csv.register_dialect – assosier dialekt med navn
• csv.writer – skriv data til en csv-fil
• csv.unregister_dialect – slett dialekten knyttet til navnet dialektregisteret
• csv.QUOTE_ALL – Sitat alt, uansett type.
• csv.QUOTE_MINIMAL – Sitatfelt med spesialtegn
• csv.QUOTE_NONNUMERIC – Oppgi alle felt som ikke er tallverdier
• csv.QUOTE_NONE – Ikke siter noe i utdata

I denne opplæringen skal vi kun fokusere på leser- og skrivefunksjonene som lar deg redigere, endre og manipulere dataene i en CSV-fil.

#import necessary modules
import csv
with open('X:\data.csv','rt')as f:
  data = csv.reader(f)
  for row in data:
        print(row)

['Programming language; Designed by; Appeared; Extension']
['Python; Guido van Rossum; 1991; .py']
['Java; James Gosling; 1995; .java']
['C++; Bjarne Stroustrup;1983;.cpp']

Hvordan lese en CSV-fil inn i en ordbok i Python

Du kan også bruke DictReader til å lese CSV-filer. Resultatene tolkes som en ordbok der overskriftsraden er nøkkelen, og andre rader er verdier.

Tenk på følgende kode

#import necessary modules
import csv

reader = csv.DictReader(open("file2.csv"))
for raw in reader:
    print(raw)

Resultatet av denne koden er:

OrderedDict([('Programming language', 'Python'), ('Designed by', 'Guido van Rossum'), (' Appeared', ' 1991'), (' Extension', ' .py')])
OrderedDict([('Programming language', 'Java'), ('Designed by', 'James Gosling'), (' Appeared', ' 1995'), (' Extension', ' .java')])
OrderedDict([('Programming language', 'C++'), ('Designed by', ' Bjarne Stroustrup'), (' Appeared', ' 1985'), (' Extension', ' .cpp')])

Og denne måten å lese data fra CSV-fil er mye enklere enn tidligere metode. Dette er imidlertid ikke den beste måten å lese data på.

#import necessary modules
import csv

with open('X:\writeData.csv', mode='w') as file:
    writer = csv.writer(file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)

    #way to write to csv file
    writer.writerow(['Programming language', 'Designed by', 'Appeared', 'Extension'])
    writer.writerow(['Python', 'Guido van Rossum', '1991', '.py'])
    writer.writerow(['Java', 'James Gosling', '1995', '.java'])
    writer.writerow(['C++', 'Bjarne Stroustrup', '1985', '.cpp'])

Programming language, Designed by, Appeared, Extension 
Python, Guido van Rossum, 1991, .py
Java, James Gosling, 1995, .java
C++, Bjarne Stroustrup,1983,.cpp

Les CSV-fil med Pandas

Pandas er et åpen kildekode-bibliotek som lar deg importere CSV inn Python og utføre datamanipulering. Pandaer gir en enkel måte å lage, manipulere og slette dataene på.

Du må installere pandas-biblioteket med kommandoen pip install pandas . I Windows, vil du utføre denne kommandoen i ledeteksten mens du er i Linux i terminalen.

Å lese CSV-en inn i en pandas DataFrame er veldig raskt og enkelt:

#import necessary modules
import pandas
result = pandas.read_csv('X:\data.csv')
print(result)

Resultatet av les CSV Pandas eksempel:

Programming language, Designed by, Appeared, Extension 
0    Python, Guido van Rossum, 1991, .py
1    Java, James Gosling, 1995, .java
2    C++, Bjarne Stroustrup,1983,.cpp

Veldig nyttig bibliotek. På bare tre linjer med kode får du samme resultat som tidligere. Pandaer vet at den første linjen i CSV-en inneholdt kolonnenavn, og den vil bruke dem automatisk.

Skriv CSV-fil med Pandas

Å skrive til CSV-fil med Pandas er like enkelt som å lese. Her kan du overbevise i det. Først må du lage DataFrame basert på følgende Python skrive til CSV-kode.

from pandas import DataFrame
C = {'Programming language': ['Python','Java', 'C++'],
        'Designed by': ['Guido van Rossum', 'James Gosling', 'Bjarne Stroustrup'],
        'Appeared': ['1991', '1995', '1985'],
        'Extension': ['.py', '.java', '.cpp'],
    }
df = DataFrame(C, columns= ['Programming language', 'Designed by', 'Appeared', 'Extension'])
export_csv = df.to_csv (r'X:\pandaresult.csv', index = None, header=True) # here you have to write path, where result file will be stored
print (df)

Her er utgangen

Programming language, Designed by, Appeared, Extension
0    Python, Guido van Rossum, 1991, .py
1    Java, James Gosling, 1995, .java
2    C++, Bjarne Stroustrup,1983,.cpp

Og CSV-filen opprettes på det angitte stedet.

Spørsmål og svar