Scrierea fișierelor CSV în Python

În acest tutorial, vom învăța să scriem fișiere CSV cu diferite formate în Python cu ajutorul unor exemple.

Vom folosi exclusiv csvmodulul încorporat în Python pentru această sarcină. Dar mai întâi, va trebui să importăm modulul ca:

 import csv 

Am acoperit deja noțiunile de bază despre modul de utilizare a csvmodulului pentru citirea și scrierea în fișiere CSV. Dacă nu aveți nicio idee despre utilizarea csvmodulului, consultați tutorialul nostru despre Python CSV: Citiți și scrieți fișiere CSV

Utilizarea de bază a csv.writer ()

Să vedem un exemplu de bază de utilizare csv.writer()pentru a vă reîmprospăta cunoștințele existente.

Exemplul 1: scrieți în fișiere CSV cu csv.writer ()

Să presupunem că vrem să scriem un fișier CSV cu următoarele intrări:

 SN, Nume, Contribuția 1, Linus Torvalds, Linux Kernel 2, Tim Berners-Lee, World Wide Web 3, Guido van Rossum, Python Programming 

Iată cum o facem.

 import csv with open('innovators.csv', 'w', newline='') as file: writer = csv.writer(file) writer.writerow(("SN", "Name", "Contribution")) writer.writerow((1, "Linus Torvalds", "Linux Kernel")) writer.writerow((2, "Tim Berners-Lee", "World Wide Web")) writer.writerow((3, "Guido van Rossum", "Python Programming")) 

Când rulăm programul de mai sus, un fișier innovators.csv este creat în directorul de lucru curent cu intrările date.

Aici am deschis fișierul innovators.csv în modul de scriere folosind open()funcția.

Pentru a afla mai multe despre deschiderea fișierelor în Python, vizitați: Intrare / ieșire fișier Python

Apoi, csv.writer()funcția este utilizată pentru a crea un writerobiect. writer.writerow()Funcția este apoi utilizată pentru a scrie un singur rânduri în fișierul CSV.

Exemplul 2: Scrierea mai multor rânduri cu writerows ()

Dacă trebuie să scriem conținutul listei bidimensionale într-un fișier CSV, iată cum o putem face.

 import csv row_list = (("SN", "Name", "Contribution"), (1, "Linus Torvalds", "Linux Kernel"), (2, "Tim Berners-Lee", "World Wide Web"), (3, "Guido van Rossum", "Python Programming")) with open('protagonist.csv', 'w', newline='') as file: writer = csv.writer(file) writer.writerows(row_list) 

Ieșirea programului este aceeași ca în Exemplul 1 .

Aici, lista noastră bidimensională este transmisă writer.writerows()funcției de a scrie conținutul listei în fișierul CSV.

Acum să vedem cum putem scrie fișiere CSV în diferite formate. Vom învăța apoi cum să personalizăm csv.writer()funcția pentru a le scrie.

Fișiere CSV cu delimitatori personalizați

În mod implicit, o virgulă este utilizată ca delimitator într-un fișier CSV. Cu toate acestea, unele fișiere CSV pot utiliza altele decât o virgulă. Puțini dintre cei populari sunt |și .

Să presupunem că dorim să îl folosim |ca delimitator în fișierul innovators.csv din Exemplul 1 . Pentru a scrie acest fișier, putem transmite un delimiterparametru suplimentar csv.writer()funcției.

Să luăm un exemplu.

Exemplul 3: Scrieți fișierul CSV având un delimitator de țeavă

 import csv data_list = (("SN", "Name", "Contribution"), (1, "Linus Torvalds", "Linux Kernel"), (2, "Tim Berners-Lee", "World Wide Web"), (3, "Guido van Rossum", "Python Programming")) with open('innovators.csv', 'w', newline='') as file: writer = csv.writer(file, delimiter='|') writer.writerows(data_list) 

Ieșire

 SN | Nume | Contribuție 1 | Linus Torvalds | Linux Kernel 2 | Tim Berners-Lee | World Wide Web 3 | Guido van Rossum | Programare Python 

După cum putem vedea, parametrul opțional delimiter = '|'ajută la specificarea writerobiectului pe care fișierul CSV ar trebui să îl aibă |ca delimitator.

Fișiere CSV cu Cotații

Unele fișiere CSV au ghilimele în jurul fiecărei sau a unor intrări.

Să luăm quotes.csv ca exemplu, cu următoarele intrări:

 „SN”; „Nume”; „Citate” 1; „Buddha”; „Ce credem că devenim” 2; „Mark Twain”; „Nu regreta niciodată nimic din ceea ce te-a făcut să zâmbești” 3; „Oscar Wilde”; „Fii tu însuți” oricine altcineva este deja luat" 

Utilizarea csv.writer()implicită nu va adăuga aceste ghilimele la intrări.

Pentru a le adăuga, va trebui să folosim un alt parametru opțional numit quoting.

Să luăm un exemplu despre modul în care citarea poate fi utilizată în jurul valorilor nenumerice și ;ca delimitatori.

Exemplul 4: Scrieți fișiere CSV cu ghilimele

 import csv row_list = ( ("SN", "Name", "Quotes"), (1, "Buddha", "What we think we become"), (2, "Mark Twain", "Never regret anything that made you smile"), (3, "Oscar Wilde", "Be yourself everyone else is already taken") ) with open('quotes.csv', 'w', newline='') as file: writer = csv.writer(file, quoting=csv.QUOTE_NONNUMERIC, delimiter=';') writer.writerows(row_list) 

Ieșire

 „SN”; „Nume”; „Citate” 1; „Buddha”; „Ce credem că devenim” 2; „Mark Twain”; „Nu regreta niciodată nimic din ceea ce te-a făcut să zâmbești” 3; „Oscar Wilde”; „Fii tu însuți” oricine altcineva este deja luat" 

Aici, fișierul quotes.csv este creat în directorul de lucru cu intrările de mai sus.

După cum puteți vedea, am trecut csv.QUOTE_NONNUMERICla quotingparametru. Este o constantă definită de csvmodul.

csv.QUOTE_NONNUMERICspecifică writerobiectul pe care ghilimelele trebuie adăugate în jurul intrărilor nenumerice.

Există alte 3 constante predefinite pe care le puteți trece la quotingparametru:

  • csv.QUOTE_ALL- Specifică writerobiectul pentru a scrie fișierul CSV cu ghilimele în jurul tuturor intrărilor.
  • csv.QUOTE_MINIMAL- Specifică writerobiectul pentru a cita doar acele câmpuri care conțin caractere speciale ( delimitator , quotechar sau orice caractere în lineterminator )
  • csv.QUOTE_NONE- Specifică writerobiectul că niciuna dintre intrări nu trebuie citată. Este valoarea implicită.

Fișiere CSV cu caracter personalizat de citare

We can also write CSV files with custom quoting characters. For that, we will have to use an optional parameter called quotechar.

Let's take an example of writing quotes.csv file in Example 4, but with * as the quoting character.

Example 5: Writing CSV files with custom quoting character

 import csv row_list = ( ("SN", "Name", "Quotes"), (1, "Buddha", "What we think we become"), (2, "Mark Twain", "Never regret anything that made you smile"), (3, "Oscar Wilde", "Be yourself everyone else is already taken") ) with open('quotes.csv', 'w', newline='') as file: writer = csv.writer(file, quoting=csv.QUOTE_NONNUMERIC, delimiter=';', quotechar='*') writer.writerows(row_list) 

Output

 *SN*;*Name*;*Quotes* 1;*Buddha*;*What we think we become* 2;*Mark Twain*;*Never regret anything that made you smile* 3;*Oscar Wilde*;*Be yourself everyone else is already taken* 

Here, we can see that quotechar='*' parameter instructs the writer object to use * as quote for all non-numeric values.

Dialects in CSV module

Notice in Example 5 that we have passed multiple parameters (quoting, delimiter and quotechar) to the csv.writer() function.

This practice is acceptable when dealing with one or two files. But it will make the code more redundant and ugly once we start working with multiple CSV files with similar formats.

As a solution to this, the csv module offers dialect as an optional parameter.

Dialect helps in grouping together many specific formatting patterns like delimiter, skipinitialspace, quoting, escapechar into a single dialect name.

It can then be passed as a parameter to multiple writer or reader instances.

Example 6: Write CSV file using dialect

Suppose we want to write a CSV file (office.csv) with the following content:

 "ID"|"Name"|"Email" "A878"|"Alfonso K. Hamby"|"[email protected]" "F854"|"Susanne Briard"|"[email protected]" "E833"|"Katja Mauer"|"[email protected]" 

The CSV file has quotes around each entry and uses | as a delimiter.

Instead of passing two individual formatting patterns, let's look at how to use dialects to write this file.

 import csv row_list = ( ("ID", "Name", "Email"), ("A878", "Alfonso K. Hamby", "[email protected]"), ("F854", "Susanne Briard", "[email protected]"), ("E833", "Katja Mauer", "[email protected]") ) csv.register_dialect('myDialect', delimiter='|', quoting=csv.QUOTE_ALL) with open('office.csv', 'w', newline='') as file: writer = csv.writer(file, dialect='myDialect') writer.writerows(row_list) 

Output

 "ID"|"Name"|"Email" "A878"|"Alfonso K. Hamby"|"[email protected]" "F854"|"Susanne Briard"|"[email protected]" "E833"|"Katja Mauer"|"[email protected]" 

Here, office.csv is created in the working directory with the above contents.

From this example, we can see that the csv.register_dialect() function is used to define a custom dialect. Its syntax is:

 csv.register_dialect(name(, dialect(, **fmtparams))) 

The custom dialect requires a name in the form of a string. Other specifications can be done either by passing a sub-class of the Dialect class, or by individual formatting patterns as shown in the example.

While creating the writer object, we pass dialect='myDialect' to specify that the writer instance must use that particular dialect.

The advantage of using dialect is that it makes the program more modular. Notice that we can reuse myDialect to write other CSV files without having to re-specify the CSV format.

Write CSV files with csv.DictWriter()

The objects of csv.DictWriter() class can be used to write to a CSV file from a Python dictionary.

The minimal syntax of the csv.DictWriter() class is:

 csv.DictWriter(file, fieldnames) 

Here,

  • file - CSV file where we want to write to
  • fieldnames - a list object which should contain the column headers specifying the order in which data should be written in the CSV file

Example 7: Python csv.DictWriter()

 import csv with open('players.csv', 'w', newline='') as file: fieldnames = ('player_name', 'fide_rating') writer = csv.DictWriter(file, fieldnames=fieldnames) writer.writeheader() writer.writerow(('player_name': 'Magnus Carlsen', 'fide_rating': 2870)) writer.writerow(('player_name': 'Fabiano Caruana', 'fide_rating': 2822)) writer.writerow(('player_name': 'Ding Liren', 'fide_rating': 2801)) 

Output

The program creates a players.csv file with the following entries:

 player_name,fide_rating Magnus Carlsen,2870 Fabiano Caruana,2822 Ding Liren,2801 

The full syntax of the csv.DictWriter() class is:

 csv.DictWriter(f, fieldnames, restval='', extrasaction='raise', dialect='excel', *args, **kwds) 

To learn more about it in detail, visit: Python csv.DictWriter() class

CSV files with lineterminator

A lineterminator is a string used to terminate lines produced by writer objects. The default value is . You can change its value by passing any string as a lineterminator parameter.

However, the reader object only recognizes or as lineterminator values. So using other characters as line terminators is highly discouraged.

doublequote & escapechar in CSV module

In order to separate delimiter characters in the entries, the csv module by default quotes the entries using quotation marks.

So, if you had an entry: He is a strong, healthy man, it will be written as: "He is a strong, healthy man".

Similarly, the csv module uses double quotes in order to escape the quote character present in the entries by default.

If you had an entry: Go to "programiz.com", it would be written as: "Go to ""programiz.com""".

Here, we can see that each " is followed by a " to escape the previous one.

doublequote

It handles how quotechar present in the entry themselves are quoted. When True, the quoting character is doubled and when False, the escapechar is used as a prefix to the quotechar. By default its value is True.

escapechar

escapechar parameter is a string to escape the delimiter if quoting is set to csv.QUOTE_NONE and quotechar if doublequote is False. Its default value is None.

Example 8: Using escapechar in csv writer

 import csv row_list = ( ('Book', 'Quote'), ('Lord of the Rings', '"All we have to decide is what to do with the time that is given us."'), ('Harry Potter', '"It matters not what someone is born, but what they grow to be."') ) with open('book.csv', 'w', newline='') as file: writer = csv.writer(file, escapechar='/', quoting=csv.QUOTE_NONE) writer.writerows(row_list) 

Output

 Book,Quote Lord of the Rings,/"All we have to decide is what to do with the time that is given us./" Harry Potter,/"It matters not what someone is born/, but what they grow to be./" 

Here, we can see that / is prefix to all the " and , because we specified quoting=csv.QUOTE_NONE.

If it wasn't defined, then, the output would be:

 Book,Quote Lord of the Rings,"""All we have to decide is what to do with the time that is given us.""" Harry Potter,"""It matters not what someone is born, but what they grow to be.""" 

Since we allow quoting, the entries with special characters(" in this case) are double-quoted. The entries with delimiter are also enclosed within quote characters.(Starting and closing quote characters)

The remaining quote characters are to escape the actual " present as part of the string, so that they are not interpreted as quotechar.

Note: The csv module can also be used for other file extensions (like: .txt) as long as their contents are in proper structure.

Lectură recomandată: Citiți fișiere CSV în Python

Articole interesante...