Trabajando con archivos csv en Python

Este artículo explica cómo cargar y analizar un archivo CSV en Python.

¿Qué es un CSV? 

CSV (valores separados por comas) es un formato de archivo simple que se utiliza para almacenar datos tabulares, como una hoja de cálculo o una base de datos. Un archivo CSV almacena datos tabulares (números y texto) en texto sin formato. Cada línea del archivo es un registro de datos. Cada registro consta de uno o más campos, separados por comas. El uso de la coma como separador de campo es el origen del nombre de este formato de archivo. Para trabajar con archivos CSV en python, hay un módulo incorporado llamado csv . 

# importing csv module
import csv
  
# csv file name
filename = "aapl.csv"
  
# initializing the titles and rows list
fields = []
rows = []
  
# reading csv file
with open(filename, 'r') as csvfile:
    # creating a csv reader object
    csvreader = csv.reader(csvfile)
      
    # extracting field names through first row
    fields = next(csvreader)
  
    # extracting each data row one by one
    for row in csvreader:
        rows.append(row)
  
    # get total number of rows
    print("Total no. of rows: %d"%(csvreader.line_num))
  
# printing the field names
print('Field names are:' + ', '.join(field for field in fields))
  
# printing first 5 rows
print('\nFirst 5 rows are:\n')
for row in rows[:5]:
    # parsing each column of a row
    for col in row:
        print("%10s"%col,end=" "),
    print('\n')

# importing the csv module
import csv
  
# field names
fields = ['Name', 'Branch', 'Year', 'CGPA']
  
# data rows of csv file
rows = [ ['Nikhil', 'COE', '2', '9.0'],
        ['Sanchit', 'COE', '2', '9.1'],
        ['Aditya', 'IT', '2', '9.3'],
        ['Sagar', 'SE', '1', '9.5'],
        ['Prateek', 'MCE', '3', '7.8'],
        ['Sahil', 'EP', '2', '9.1']]
  
# name of csv file
filename = "university_records.csv"
  
# writing to csv file
with open(filename, 'w') as csvfile:
    # creating a csv writer object
    csvwriter = csv.writer(csvfile)
      
    # writing the fields
    csvwriter.writerow(fields)
      
    # writing the data rows
    csvwriter.writerows(rows)

# importing the csv module
import csv
  
# my data rows as dictionary objects
mydict =[{'branch': 'COE', 'cgpa': '9.0', 
          'name': 'Nikhil', 'year': '2'},
        {'branch': 'COE', 'cgpa': '9.1', 
         'name': 'Sanchit', 'year': '2'},
        {'branch': 'IT', 'cgpa': '9.3', 
         'name': 'Aditya', 'year': '2'},
        {'branch': 'SE', 'cgpa': '9.5', 
         'name': 'Sagar', 'year': '1'},
        {'branch': 'MCE', 'cgpa': '7.8', 
         'name': 'Prateek', 'year': '3'},
        {'branch': 'EP', 'cgpa': '9.1', 
         'name': 'Sahil', 'year': '2'}]
  
# field names
fields = ['name', 'branch', 'year', 'cgpa']
  
# name of csv file
filename = "university_records.csv"
  
# writing to csv file
with open(filename, 'w') as csvfile:
    # creating a csv dict writer object
    writer = csv.DictWriter(csvfile, fieldnames = fields)
      
    # writing headers (field names)
    writer.writeheader()
      
    # writing data rows
    writer.writerows(mydict)