Python CSV Reader Example: Ako efektívne čítať CSV súbory v Pythone

Python je jedným z najpopulárnejších programovacích jazykov na svete a to vďaka svojej jednoduchosti a flexibilite. Pre každého, kto pracuje s dátami, je nevyhnutné poznať spôsob, ako načítať a manipulovať s údajmi uloženými v CSV (Comma-Separated Values) súboroch. Dnes sa pozrieme na to, ako môžeme v Pythone načítať CSV súbory pomocou vstavaného modulu csv.

Čo je to CSV súbor?

CSV (Comma-Separated Values) je formát súborov, ktorý sa používa na ukladanie dát v tabuľkovom formáte. Každý riadok v súbore reprezentuje jeden záznam, pričom jednotlivé hodnoty sú oddelené čiarkami (alebo inými oddeľovačmi, ako sú bodkočiarky). Tento formát je veľmi populárny, pretože je jednoducho čitateľný a podporovaný väčšinou databáz a tabulkových procesorov, ako je Microsoft Excel alebo Google Sheets.

Prečo používať Python na prácu s CSV súbormi?

Python poskytuje robustný spôsob spracovania CSV súborov pomocou knižnice csv, ktorá umožňuje načítanie, zápis a spracovanie týchto súborov bez potreby externých knižníc. Knižnica csv je veľmi flexibilná a jednoducho použiteľná na manipuláciu s dátami. Ak sa potrebujete naučiť, ako efektívne čítať a spracovávať CSV súbory v Pythone, tento článok je pre vás!

Začneme so základným čítaním CSV súboru

Poďme si ukázať základný príklad, ako môžeme načítať CSV súbor a vypísať jeho obsah. Na to použijeme knižnicu csv, ktorá poskytuje jednoduché metódy na prácu s CSV súbormi.

import csv

# Otvorenie CSV súboru na čítanie
with open('priklad.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Prechádzanie jednotlivými riadkami CSV súboru
    for row in reader:
        print(row)

V tomto príklade používame funkciu csv.reader() na načítanie súboru. Riadky sú potom vypísané v podobe zoznamov, kde každá hodnota v riadku je prvok tohto zoznamu.

Pokročilé techniky: Práca s hlavičkami a slovníkmi

Ak má váš CSV súbor hlavičku (prvý riadok obsahuje názvy stĺpcov), môžete ho jednoducho spracovať pomocou knižnice csv. Pre tento účel je ideálny csv.DictReader(), ktorý načítava každý riadok ako slovník, kde kľúče sú názvy stĺpcov a hodnoty sú dáta v jednotlivých bunkách.

import csv

# Otvorenie CSV súboru s hlavičkou
with open('priklad_s_hlavickou.csv', mode='r') as file:
    reader = csv.DictReader(file)
    
    # Prechádzanie jednotlivými riadkami
    for row in reader:
        print(row)

Tento kód načíta CSV súbor a každý riadok predstavuje slovník. Názvy stĺpcov sa používajú ako kľúče a hodnoty ako hodnoty týchto kľúčov.

Filterovanie a spracovanie dát z CSV súboru

Ak chcete filtrovať alebo manipulovať s dátami priamo počas načítania CSV súboru, môžete pridať podmienky do vašich slučiek. Pozrime sa na príklad, kde budeme filtrovať riadky na základe určitého stĺpca.

import csv

# Otvorenie CSV súboru
with open('priklad_s_hlavickou.csv', mode='r') as file:
    reader = csv.DictReader(file)
    
    # Filtrovanie podľa hodnoty v stĺpci 'vek'
    for row in reader:
        if int(row['vek']) > 30:
            print(row)

V tomto príklade čítame CSV súbor a vypisujeme len tie riadky, kde hodnota v stĺpci "vek" je väčšia než 30. Týmto spôsobom môžeme efektívne filtrovať údaje počas načítania.

Ukladanie dát do nového CSV súboru

Okrem čítania dát z CSV súborov, môžeme v Pythone aj zapisovať dáta do nových CSV súborov. Na tento účel použijeme metódu csv.writer(). Tu je príklad, ako uložiť dáta do nového CSV súboru:

import csv

# Otvorenie nového CSV súboru na zápis
with open('novy_soubor.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Zápis hlavičky
    writer.writerow(['meno', 'vek', 'mesto'])
    
    # Zápis dát
    writer.writerow(['Ján', 25, 'Bratislava'])
    writer.writerow(['Eva', 32, 'Košice'])
    writer.writerow(['Martin', 28, 'Nitra'])

V tomto príklade zapisujeme údaje do nového CSV súboru. Prvým krokom je zapisovanie hlavičky (názvov stĺpcov) a potom zapisovanie jednotlivých riadkov s dátami.

Príklady: Použitie CSV na analýzu údajov

CSV súbory sa veľmi často používajú pri analýze údajov. Ak máte veľký CSV súbor s rôznymi informáciami, môžete ho spracovať a analyzovať pomocou Pythonu. Tu je jednoduchý príklad, ako z CSV súboru získať priemerný vek osôb.

import csv

# Otvorenie CSV súboru na čítanie
with open('priklad_s_hlavickou.csv', mode='r') as file:
    reader = csv.DictReader(file)
    
    total_age = 0
    count = 0
    
    # Prechádzanie jednotlivými riadkami a počítanie priemerného veku
    for row in reader:
        total_age += int(row['vek'])
        count += 1
    
    average_age = total_age / count
    print(f"Priemerný vek: {average_age}")

Tento príklad načítava CSV súbor, počíta celkový vek a potom vypočíta priemerný vek všetkých osôb v súbore.

Záver

Práca s CSV súbormi v Pythone je veľmi jednoduchá a flexibilná. Pomocou knižnice csv môžete efektívne čítať, zapisovať a spracovávať údaje zo súborov v CSV formáte. Dúfame, že tento článok vám pomohol pochopiť, ako pracovať s CSV súbormi v Pythone, a poskytol vám niekoľko užitočných príkladov, ktoré môžete využiť vo svojich projektoch!