CSV-bestanden schrijven in Python

Inhoudsopgave

Basisgebruik van csv.writer ()
CSV-bestanden met aangepaste scheidingstekens
CSV-bestanden met offertes
CSV-bestanden met een aangepast aanhalingsteken
Dialects in CSV module
Write CSV files with csv.DictWriter()
CSV files with lineterminator
doublequote & escapechar in CSV module

In deze tutorial leren we met behulp van voorbeelden CSV-bestanden met verschillende formaten in Python te schrijven.

We gaan csvvoor deze taak exclusief de module gebruiken die in Python is ingebouwd. Maar eerst zullen we de module moeten importeren als:

 import csv

We hebben al de basisprincipes besproken van het gebruik van de csvmodule om CSV-bestanden te lezen en erin te schrijven. Als je geen idee hebt over het gebruik van de csvmodule, bekijk dan onze tutorial over Python CSV: CSV-bestanden lezen en schrijven

Basisgebruik van csv.writer ()

Laten we eens kijken naar een eenvoudig voorbeeld van het gebruik csv.writer()om uw bestaande kennis op te frissen.

Voorbeeld 1: schrijf naar CSV-bestanden met csv.writer ()

Stel dat we een CSV-bestand willen schrijven met de volgende vermeldingen:

 SN, naam, bijdrage 1, Linus Torvalds, Linux Kernel 2, Tim Berners-Lee, World Wide Web 3, Guido van Rossum, Python Programming

Hier is hoe we het doen.

 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"))

Wanneer we het bovenstaande programma uitvoeren, wordt een innovators.csv- bestand gemaakt in de huidige werkdirectory met de opgegeven vermeldingen.

Hier hebben we het innovators.csv- bestand in schrijfmodus geopend met behulp van open()function.

Ga voor meer informatie over het openen van bestanden in Python naar: Python File Input / Output

Vervolgens wordt de csv.writer()functie gebruikt om een writerobject te maken . De writer.writerow()functie wordt vervolgens gebruikt om enkele rijen naar het CSV-bestand te schrijven.

Voorbeeld 2: Meerdere rijen schrijven met writerows ()

Als we de inhoud van de tweedimensionale lijst naar een CSV-bestand moeten schrijven, kunnen we dat als volgt doen.

 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)

De output van het programma is hetzelfde als in Voorbeeld 1 .

Hier wordt onze 2-dimensionale lijst doorgegeven aan de writer.writerows()functie om de inhoud van de lijst naar het CSV-bestand te schrijven.

Laten we nu eens kijken hoe we CSV-bestanden in verschillende formaten kunnen schrijven. We zullen dan leren hoe we de csv.writer()functie kunnen aanpassen om ze te schrijven.

CSV-bestanden met aangepaste scheidingstekens

Standaard wordt een komma gebruikt als scheidingsteken in een CSV-bestand. Sommige CSV-bestanden kunnen echter andere scheidingstekens gebruiken dan een komma. Er zijn |maar weinig populaire zijn en .

Stel dat we |als scheidingsteken willen gebruiken in het innovators.csv- bestand van Voorbeeld 1 . Om dit bestand te schrijven, kunnen we een extra delimiterparameter aan de csv.writer()functie doorgeven .

Laten we een voorbeeld nemen.

Voorbeeld 3: CSV-bestand schrijven met pijpscheidingsteken

 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)

Uitvoer

 SN | Naam | Bijdrage 1 | Linus Torvalds | Linux Kernel 2 | Tim Berners-Lee | World Wide Web 3 | Guido van Rossum | Python Programming

Zoals we kunnen zien, delimiter = '|'helpt de optionele parameter bij het specificeren van het writerobject dat het CSV-bestand |als scheidingsteken moet hebben .

CSV-bestanden met offertes

Sommige CSV-bestanden hebben aanhalingstekens rond elk of enkele items.

Laten we als voorbeeld quotes.csv nemen , met de volgende vermeldingen:

 "SN"; "Naam"; "Quotes" 1; "Boeddha"; "Wat we denken dat we worden" 2; "Mark Twain"; "Nooit spijt hebben van iets dat je deed glimlachen" 3; "Oscar Wilde"; "Wees jezelf de rest is al bezet"

Als u csv.writer()standaard gebruikt, worden deze aanhalingstekens niet aan de vermeldingen toegevoegd.

Om ze toe te voegen, zullen we een andere optionele parameter moeten gebruiken genaamd quoting.

Laten we een voorbeeld nemen van hoe aanhalingstekens kunnen worden gebruikt rond de niet-numerieke waarden en ;als scheidingstekens.

Voorbeeld 4: schrijf CSV-bestanden met aanhalingstekens

 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)

Uitvoer

 "SN"; "Naam"; "Quotes" 1; "Boeddha"; "Wat we denken dat we worden" 2; "Mark Twain"; "Nooit spijt hebben van iets dat je deed glimlachen" 3; "Oscar Wilde"; "Wees jezelf de rest is al bezet"

Hier wordt het quotes.csv- bestand gemaakt in de werkmap met de bovenstaande vermeldingen.

Zoals u kunt zien, zijn we doorgegaan csv.QUOTE_NONNUMERICnaar de quotingparameter. Het is een constante gedefinieerd door de csvmodule.

csv.QUOTE_NONNUMERICspecificeert het writerobject dat aanhalingstekens moeten worden toegevoegd rond de niet-numerieke items.

Er zijn 3 andere vooraf gedefinieerde constanten die u aan de quotingparameter kunt doorgeven :

csv.QUOTE_ALL- Specificeert het writerobject om een CSV-bestand te schrijven met aanhalingstekens rond alle vermeldingen.
csv.QUOTE_MINIMAL- Specificeert het writerobject om alleen die velden aan te halen die speciale tekens bevatten ( scheidingsteken , quotechar of andere tekens in lineterminator )
csv.QUOTE_NONE- Specificeert het writerobject dat geen van de items moet worden geciteerd. Dit is de standaardwaarde.

CSV-bestanden met een aangepast aanhalingsteken

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.

Aanbevolen literatuur: lees CSV-bestanden in Python