Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 1 | :mod:`csv` --- CSV File Reading and Writing |
| 2 | =========================================== |
| 3 | |
| 4 | .. module:: csv |
| 5 | :synopsis: Write and read tabular data to and from delimited files. |
| 6 | .. sectionauthor:: Skip Montanaro <skip@pobox.com> |
| 7 | |
| 8 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 9 | .. index:: |
| 10 | single: csv |
| 11 | pair: data; tabular |
| 12 | |
| 13 | The so-called CSV (Comma Separated Values) format is the most common import and |
Skip Montanaro | b40dea7 | 2011-03-19 09:09:30 -0500 | [diff] [blame] | 14 | export format for spreadsheets and databases. CSV format was used for many |
| 15 | years prior to attempts to describe the format in a standardized way in |
| 16 | :rfc:`4180`. The lack of a well-defined standard means that subtle differences |
| 17 | often exist in the data produced and consumed by different applications. These |
| 18 | differences can make it annoying to process CSV files from multiple sources. |
| 19 | Still, while the delimiters and quoting characters vary, the overall format is |
| 20 | similar enough that it is possible to write a single module which can |
| 21 | efficiently manipulate such data, hiding the details of reading and writing the |
| 22 | data from the programmer. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 23 | |
| 24 | The :mod:`csv` module implements classes to read and write tabular data in CSV |
| 25 | format. It allows programmers to say, "write this data in the format preferred |
| 26 | by Excel," or "read data from this file which was generated by Excel," without |
| 27 | knowing the precise details of the CSV format used by Excel. Programmers can |
| 28 | also describe the CSV formats understood by other applications or define their |
| 29 | own special-purpose CSV formats. |
| 30 | |
| 31 | The :mod:`csv` module's :class:`reader` and :class:`writer` objects read and |
| 32 | write sequences. Programmers can also read and write data in dictionary form |
| 33 | using the :class:`DictReader` and :class:`DictWriter` classes. |
| 34 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 35 | .. seealso:: |
| 36 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 37 | :pep:`305` - CSV File API |
| 38 | The Python Enhancement Proposal which proposed this addition to Python. |
| 39 | |
| 40 | |
| 41 | .. _csv-contents: |
| 42 | |
| 43 | Module Contents |
| 44 | --------------- |
| 45 | |
| 46 | The :mod:`csv` module defines the following functions: |
| 47 | |
| 48 | |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 49 | .. function:: reader(csvfile, dialect='excel', **fmtparams) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 50 | |
| 51 | Return a reader object which will iterate over lines in the given *csvfile*. |
Georg Brandl | 9afde1c | 2007-11-01 20:32:30 +0000 | [diff] [blame] | 52 | *csvfile* can be any object which supports the :term:`iterator` protocol and returns a |
Georg Brandl | b30f330 | 2011-01-06 09:23:56 +0000 | [diff] [blame] | 53 | string each time its :meth:`!__next__` method is called --- :term:`file objects |
Antoine Pitrou | 11cb961 | 2010-09-15 11:11:28 +0000 | [diff] [blame] | 54 | <file object>` and list objects are both suitable. If *csvfile* is a file object, |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 55 | it should be opened with ``newline=''``. [1]_ An optional |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 56 | *dialect* parameter can be given which is used to define a set of parameters |
| 57 | specific to a particular CSV dialect. It may be an instance of a subclass of |
| 58 | the :class:`Dialect` class or one of the strings returned by the |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 59 | :func:`list_dialects` function. The other optional *fmtparams* keyword arguments |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 60 | can be given to override individual formatting parameters in the current |
| 61 | dialect. For full details about the dialect and formatting parameters, see |
| 62 | section :ref:`csv-fmt-params`. |
| 63 | |
Skip Montanaro | 0468df3 | 2009-03-25 00:52:18 +0000 | [diff] [blame] | 64 | Each row read from the csv file is returned as a list of strings. No |
R. David Murray | 8b7d4aa | 2009-04-04 01:38:38 +0000 | [diff] [blame] | 65 | automatic data type conversion is performed unless the ``QUOTE_NONNUMERIC`` format |
| 66 | option is specified (in which case unquoted fields are transformed into floats). |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 67 | |
Christian Heimes | b9eccbf | 2007-12-05 20:18:38 +0000 | [diff] [blame] | 68 | A short usage example:: |
Georg Brandl | 48310cd | 2009-01-03 21:18:54 +0000 | [diff] [blame] | 69 | |
Christian Heimes | b9eccbf | 2007-12-05 20:18:38 +0000 | [diff] [blame] | 70 | >>> import csv |
R. David Murray | 8b7d4aa | 2009-04-04 01:38:38 +0000 | [diff] [blame] | 71 | >>> spamReader = csv.reader(open('eggs.csv', newline=''), delimiter=' ', quotechar='|') |
Christian Heimes | b9eccbf | 2007-12-05 20:18:38 +0000 | [diff] [blame] | 72 | >>> for row in spamReader: |
Georg Brandl | f694518 | 2008-02-01 11:56:49 +0000 | [diff] [blame] | 73 | ... print(', '.join(row)) |
Christian Heimes | b9eccbf | 2007-12-05 20:18:38 +0000 | [diff] [blame] | 74 | Spam, Spam, Spam, Spam, Spam, Baked Beans |
| 75 | Spam, Lovely Spam, Wonderful Spam |
| 76 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 77 | |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 78 | .. function:: writer(csvfile, dialect='excel', **fmtparams) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 79 | |
| 80 | Return a writer object responsible for converting the user's data into delimited |
| 81 | strings on the given file-like object. *csvfile* can be any object with a |
R David Murray | 9c0d5ea | 2011-03-20 11:18:21 -0400 | [diff] [blame] | 82 | :func:`write` method. If *csvfile* is a file object, it should be opened with |
| 83 | ``newline=''`` [1]_. An optional *dialect* |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 84 | parameter can be given which is used to define a set of parameters specific to a |
| 85 | particular CSV dialect. It may be an instance of a subclass of the |
| 86 | :class:`Dialect` class or one of the strings returned by the |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 87 | :func:`list_dialects` function. The other optional *fmtparams* keyword arguments |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 88 | can be given to override individual formatting parameters in the current |
| 89 | dialect. For full details about the dialect and formatting parameters, see |
| 90 | section :ref:`csv-fmt-params`. To make it |
| 91 | as easy as possible to interface with modules which implement the DB API, the |
| 92 | value :const:`None` is written as the empty string. While this isn't a |
| 93 | reversible transformation, it makes it easier to dump SQL NULL data values to |
| 94 | CSV files without preprocessing the data returned from a ``cursor.fetch*`` call. |
| 95 | All other non-string data are stringified with :func:`str` before being written. |
| 96 | |
Christian Heimes | b9eccbf | 2007-12-05 20:18:38 +0000 | [diff] [blame] | 97 | A short usage example:: |
| 98 | |
| 99 | >>> import csv |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 100 | >>> spamWriter = csv.writer(open('eggs.csv', 'w', newline=''), delimiter=' ', |
Skip Montanaro | f290cb5 | 2009-08-18 14:37:42 +0000 | [diff] [blame] | 101 | ... quotechar='|', quoting=csv.QUOTE_MINIMAL) |
Christian Heimes | b9eccbf | 2007-12-05 20:18:38 +0000 | [diff] [blame] | 102 | >>> spamWriter.writerow(['Spam'] * 5 + ['Baked Beans']) |
| 103 | >>> spamWriter.writerow(['Spam', 'Lovely Spam', 'Wonderful Spam']) |
| 104 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 105 | |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 106 | .. function:: register_dialect(name[, dialect], **fmtparams) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 107 | |
Georg Brandl | f694518 | 2008-02-01 11:56:49 +0000 | [diff] [blame] | 108 | Associate *dialect* with *name*. *name* must be a string. The |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 109 | dialect can be specified either by passing a sub-class of :class:`Dialect`, or |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 110 | by *fmtparams* keyword arguments, or both, with keyword arguments overriding |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 111 | parameters of the dialect. For full details about the dialect and formatting |
| 112 | parameters, see section :ref:`csv-fmt-params`. |
| 113 | |
| 114 | |
| 115 | .. function:: unregister_dialect(name) |
| 116 | |
| 117 | Delete the dialect associated with *name* from the dialect registry. An |
| 118 | :exc:`Error` is raised if *name* is not a registered dialect name. |
| 119 | |
| 120 | |
| 121 | .. function:: get_dialect(name) |
| 122 | |
Georg Brandl | 6554cb9 | 2007-12-02 23:15:43 +0000 | [diff] [blame] | 123 | Return the dialect associated with *name*. An :exc:`Error` is raised if |
| 124 | *name* is not a registered dialect name. This function returns an immutable |
| 125 | :class:`Dialect`. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 126 | |
| 127 | .. function:: list_dialects() |
| 128 | |
| 129 | Return the names of all registered dialects. |
| 130 | |
| 131 | |
| 132 | .. function:: field_size_limit([new_limit]) |
| 133 | |
| 134 | Returns the current maximum field size allowed by the parser. If *new_limit* is |
| 135 | given, this becomes the new limit. |
| 136 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 137 | |
| 138 | The :mod:`csv` module defines the following classes: |
| 139 | |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 140 | .. class:: DictReader(csvfile, fieldnames=None, restkey=None, restval=None, dialect='excel', *args, **kwds) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 141 | |
| 142 | Create an object which operates like a regular reader but maps the information |
| 143 | read into a dict whose keys are given by the optional *fieldnames* parameter. |
| 144 | If the *fieldnames* parameter is omitted, the values in the first row of the |
R. David Murray | 578ec90 | 2009-11-09 14:21:38 +0000 | [diff] [blame] | 145 | *csvfile* will be used as the fieldnames. If the row read has more fields |
| 146 | than the fieldnames sequence, the remaining data is added as a sequence |
| 147 | keyed by the value of *restkey*. If the row read has fewer fields than the |
| 148 | fieldnames sequence, the remaining keys take the value of the optional |
| 149 | *restval* parameter. Any other optional or keyword arguments are passed to |
| 150 | the underlying :class:`reader` instance. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 151 | |
| 152 | |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 153 | .. class:: DictWriter(csvfile, fieldnames, restval='', extrasaction='raise', dialect='excel', *args, **kwds) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 154 | |
| 155 | Create an object which operates like a regular writer but maps dictionaries onto |
| 156 | output rows. The *fieldnames* parameter identifies the order in which values in |
| 157 | the dictionary passed to the :meth:`writerow` method are written to the |
| 158 | *csvfile*. The optional *restval* parameter specifies the value to be written |
| 159 | if the dictionary is missing a key in *fieldnames*. If the dictionary passed to |
| 160 | the :meth:`writerow` method contains a key not found in *fieldnames*, the |
| 161 | optional *extrasaction* parameter indicates what action to take. If it is set |
| 162 | to ``'raise'`` a :exc:`ValueError` is raised. If it is set to ``'ignore'``, |
| 163 | extra values in the dictionary are ignored. Any other optional or keyword |
| 164 | arguments are passed to the underlying :class:`writer` instance. |
| 165 | |
| 166 | Note that unlike the :class:`DictReader` class, the *fieldnames* parameter of |
| 167 | the :class:`DictWriter` is not optional. Since Python's :class:`dict` objects |
| 168 | are not ordered, there is not enough information available to deduce the order |
| 169 | in which the row should be written to the *csvfile*. |
| 170 | |
| 171 | |
| 172 | .. class:: Dialect |
| 173 | |
| 174 | The :class:`Dialect` class is a container class relied on primarily for its |
| 175 | attributes, which are used to define the parameters for a specific |
| 176 | :class:`reader` or :class:`writer` instance. |
| 177 | |
| 178 | |
| 179 | .. class:: excel() |
| 180 | |
| 181 | The :class:`excel` class defines the usual properties of an Excel-generated CSV |
| 182 | file. It is registered with the dialect name ``'excel'``. |
| 183 | |
| 184 | |
| 185 | .. class:: excel_tab() |
| 186 | |
| 187 | The :class:`excel_tab` class defines the usual properties of an Excel-generated |
| 188 | TAB-delimited file. It is registered with the dialect name ``'excel-tab'``. |
| 189 | |
| 190 | |
Georg Brandl | 7424dd3 | 2010-10-27 07:27:06 +0000 | [diff] [blame] | 191 | .. class:: unix_dialect() |
| 192 | |
| 193 | The :class:`unix_dialect` class defines the usual properties of a CSV file |
| 194 | generated on UNIX systems, i.e. using ``'\n'`` as line terminator and quoting |
| 195 | all fields. It is registered with the dialect name ``'unix'``. |
| 196 | |
| 197 | .. versionadded:: 3.2 |
| 198 | |
| 199 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 200 | .. class:: Sniffer() |
| 201 | |
| 202 | The :class:`Sniffer` class is used to deduce the format of a CSV file. |
| 203 | |
Benjamin Peterson | e41251e | 2008-04-25 01:59:09 +0000 | [diff] [blame] | 204 | The :class:`Sniffer` class provides two methods: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 205 | |
Georg Brandl | c2a4f4f | 2009-04-10 09:03:43 +0000 | [diff] [blame] | 206 | .. method:: sniff(sample, delimiters=None) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 207 | |
Benjamin Peterson | e41251e | 2008-04-25 01:59:09 +0000 | [diff] [blame] | 208 | Analyze the given *sample* and return a :class:`Dialect` subclass |
| 209 | reflecting the parameters found. If the optional *delimiters* parameter |
| 210 | is given, it is interpreted as a string containing possible valid |
| 211 | delimiter characters. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 212 | |
| 213 | |
Benjamin Peterson | e41251e | 2008-04-25 01:59:09 +0000 | [diff] [blame] | 214 | .. method:: has_header(sample) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 215 | |
Benjamin Peterson | e41251e | 2008-04-25 01:59:09 +0000 | [diff] [blame] | 216 | Analyze the sample text (presumed to be in CSV format) and return |
| 217 | :const:`True` if the first row appears to be a series of column headers. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 218 | |
Christian Heimes | 7f04431 | 2008-01-06 17:05:40 +0000 | [diff] [blame] | 219 | An example for :class:`Sniffer` use:: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 220 | |
Christian Heimes | 7f04431 | 2008-01-06 17:05:40 +0000 | [diff] [blame] | 221 | csvfile = open("example.csv") |
| 222 | dialect = csv.Sniffer().sniff(csvfile.read(1024)) |
| 223 | csvfile.seek(0) |
| 224 | reader = csv.reader(csvfile, dialect) |
| 225 | # ... process CSV file contents here ... |
| 226 | |
| 227 | |
| 228 | The :mod:`csv` module defines the following constants: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 229 | |
| 230 | .. data:: QUOTE_ALL |
| 231 | |
| 232 | Instructs :class:`writer` objects to quote all fields. |
| 233 | |
| 234 | |
| 235 | .. data:: QUOTE_MINIMAL |
| 236 | |
| 237 | Instructs :class:`writer` objects to only quote those fields which contain |
| 238 | special characters such as *delimiter*, *quotechar* or any of the characters in |
| 239 | *lineterminator*. |
| 240 | |
| 241 | |
| 242 | .. data:: QUOTE_NONNUMERIC |
| 243 | |
| 244 | Instructs :class:`writer` objects to quote all non-numeric fields. |
| 245 | |
| 246 | Instructs the reader to convert all non-quoted fields to type *float*. |
| 247 | |
| 248 | |
| 249 | .. data:: QUOTE_NONE |
| 250 | |
| 251 | Instructs :class:`writer` objects to never quote fields. When the current |
| 252 | *delimiter* occurs in output data it is preceded by the current *escapechar* |
| 253 | character. If *escapechar* is not set, the writer will raise :exc:`Error` if |
| 254 | any characters that require escaping are encountered. |
| 255 | |
| 256 | Instructs :class:`reader` to perform no special processing of quote characters. |
| 257 | |
| 258 | The :mod:`csv` module defines the following exception: |
| 259 | |
| 260 | |
| 261 | .. exception:: Error |
| 262 | |
| 263 | Raised by any of the functions when an error is detected. |
| 264 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 265 | .. _csv-fmt-params: |
| 266 | |
| 267 | Dialects and Formatting Parameters |
| 268 | ---------------------------------- |
| 269 | |
| 270 | To make it easier to specify the format of input and output records, specific |
| 271 | formatting parameters are grouped together into dialects. A dialect is a |
| 272 | subclass of the :class:`Dialect` class having a set of specific methods and a |
| 273 | single :meth:`validate` method. When creating :class:`reader` or |
| 274 | :class:`writer` objects, the programmer can specify a string or a subclass of |
| 275 | the :class:`Dialect` class as the dialect parameter. In addition to, or instead |
| 276 | of, the *dialect* parameter, the programmer can also specify individual |
| 277 | formatting parameters, which have the same names as the attributes defined below |
| 278 | for the :class:`Dialect` class. |
| 279 | |
| 280 | Dialects support the following attributes: |
| 281 | |
| 282 | |
| 283 | .. attribute:: Dialect.delimiter |
| 284 | |
| 285 | A one-character string used to separate fields. It defaults to ``','``. |
| 286 | |
| 287 | |
| 288 | .. attribute:: Dialect.doublequote |
| 289 | |
| 290 | Controls how instances of *quotechar* appearing inside a field should be |
| 291 | themselves be quoted. When :const:`True`, the character is doubled. When |
| 292 | :const:`False`, the *escapechar* is used as a prefix to the *quotechar*. It |
| 293 | defaults to :const:`True`. |
| 294 | |
| 295 | On output, if *doublequote* is :const:`False` and no *escapechar* is set, |
| 296 | :exc:`Error` is raised if a *quotechar* is found in a field. |
| 297 | |
| 298 | |
| 299 | .. attribute:: Dialect.escapechar |
| 300 | |
| 301 | A one-character string used by the writer to escape the *delimiter* if *quoting* |
| 302 | is set to :const:`QUOTE_NONE` and the *quotechar* if *doublequote* is |
| 303 | :const:`False`. On reading, the *escapechar* removes any special meaning from |
| 304 | the following character. It defaults to :const:`None`, which disables escaping. |
| 305 | |
| 306 | |
| 307 | .. attribute:: Dialect.lineterminator |
| 308 | |
| 309 | The string used to terminate lines produced by the :class:`writer`. It defaults |
| 310 | to ``'\r\n'``. |
| 311 | |
| 312 | .. note:: |
| 313 | |
| 314 | The :class:`reader` is hard-coded to recognise either ``'\r'`` or ``'\n'`` as |
| 315 | end-of-line, and ignores *lineterminator*. This behavior may change in the |
| 316 | future. |
| 317 | |
| 318 | |
| 319 | .. attribute:: Dialect.quotechar |
| 320 | |
| 321 | A one-character string used to quote fields containing special characters, such |
| 322 | as the *delimiter* or *quotechar*, or which contain new-line characters. It |
| 323 | defaults to ``'"'``. |
| 324 | |
| 325 | |
| 326 | .. attribute:: Dialect.quoting |
| 327 | |
| 328 | Controls when quotes should be generated by the writer and recognised by the |
| 329 | reader. It can take on any of the :const:`QUOTE_\*` constants (see section |
| 330 | :ref:`csv-contents`) and defaults to :const:`QUOTE_MINIMAL`. |
| 331 | |
| 332 | |
| 333 | .. attribute:: Dialect.skipinitialspace |
| 334 | |
| 335 | When :const:`True`, whitespace immediately following the *delimiter* is ignored. |
| 336 | The default is :const:`False`. |
| 337 | |
| 338 | |
| 339 | Reader Objects |
| 340 | -------------- |
| 341 | |
| 342 | Reader objects (:class:`DictReader` instances and objects returned by the |
| 343 | :func:`reader` function) have the following public methods: |
| 344 | |
Georg Brandl | c748506 | 2009-04-01 15:53:15 +0000 | [diff] [blame] | 345 | .. method:: csvreader.__next__() |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 346 | |
| 347 | Return the next row of the reader's iterable object as a list, parsed according |
Georg Brandl | c748506 | 2009-04-01 15:53:15 +0000 | [diff] [blame] | 348 | to the current dialect. Usually you should call this as ``next(reader)``. |
| 349 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 350 | |
| 351 | Reader objects have the following public attributes: |
| 352 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 353 | .. attribute:: csvreader.dialect |
| 354 | |
| 355 | A read-only description of the dialect in use by the parser. |
| 356 | |
| 357 | |
| 358 | .. attribute:: csvreader.line_num |
| 359 | |
| 360 | The number of lines read from the source iterator. This is not the same as the |
| 361 | number of records returned, as records can span multiple lines. |
| 362 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 363 | |
Skip Montanaro | af8fcfa | 2008-08-09 19:44:22 +0000 | [diff] [blame] | 364 | DictReader objects have the following public attribute: |
| 365 | |
Skip Montanaro | af8fcfa | 2008-08-09 19:44:22 +0000 | [diff] [blame] | 366 | .. attribute:: csvreader.fieldnames |
| 367 | |
| 368 | If not passed as a parameter when creating the object, this attribute is |
| 369 | initialized upon first access or when the first record is read from the |
| 370 | file. |
| 371 | |
Skip Montanaro | af8fcfa | 2008-08-09 19:44:22 +0000 | [diff] [blame] | 372 | |
| 373 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 374 | Writer Objects |
| 375 | -------------- |
| 376 | |
| 377 | :class:`Writer` objects (:class:`DictWriter` instances and objects returned by |
| 378 | the :func:`writer` function) have the following public methods. A *row* must be |
| 379 | a sequence of strings or numbers for :class:`Writer` objects and a dictionary |
| 380 | mapping fieldnames to strings or numbers (by passing them through :func:`str` |
| 381 | first) for :class:`DictWriter` objects. Note that complex numbers are written |
| 382 | out surrounded by parens. This may cause some problems for other programs which |
| 383 | read CSV files (assuming they support complex numbers at all). |
| 384 | |
| 385 | |
| 386 | .. method:: csvwriter.writerow(row) |
| 387 | |
| 388 | Write the *row* parameter to the writer's file object, formatted according to |
| 389 | the current dialect. |
| 390 | |
| 391 | |
| 392 | .. method:: csvwriter.writerows(rows) |
| 393 | |
| 394 | Write all the *rows* parameters (a list of *row* objects as described above) to |
| 395 | the writer's file object, formatted according to the current dialect. |
| 396 | |
| 397 | Writer objects have the following public attribute: |
| 398 | |
| 399 | |
| 400 | .. attribute:: csvwriter.dialect |
| 401 | |
| 402 | A read-only description of the dialect in use by the writer. |
| 403 | |
| 404 | |
R. David Murray | be0698b | 2010-02-23 22:57:58 +0000 | [diff] [blame] | 405 | DictWriter objects have the following public method: |
| 406 | |
| 407 | |
| 408 | .. method:: DictWriter.writeheader() |
| 409 | |
| 410 | Write a row with the field names (as specified in the constructor). |
| 411 | |
R. David Murray | 19e4548 | 2010-02-23 23:00:34 +0000 | [diff] [blame] | 412 | .. versionadded:: 3.2 |
R. David Murray | be0698b | 2010-02-23 22:57:58 +0000 | [diff] [blame] | 413 | |
| 414 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 415 | .. _csv-examples: |
| 416 | |
| 417 | Examples |
| 418 | -------- |
| 419 | |
| 420 | The simplest example of reading a CSV file:: |
| 421 | |
| 422 | import csv |
Eli Bendersky | 6860a92 | 2011-03-11 15:47:36 +0200 | [diff] [blame] | 423 | with open('some.csv', newline='') as f: |
| 424 | reader = csv.reader(f) |
| 425 | for row in reader: |
| 426 | print(row) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 427 | |
| 428 | Reading a file with an alternate format:: |
| 429 | |
| 430 | import csv |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 431 | with open('passwd', newline='') as f: |
Eli Bendersky | 6860a92 | 2011-03-11 15:47:36 +0200 | [diff] [blame] | 432 | reader = csv.reader(f, delimiter=':', quoting=csv.QUOTE_NONE) |
| 433 | for row in reader: |
| 434 | print(row) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 435 | |
| 436 | The corresponding simplest possible writing example is:: |
| 437 | |
| 438 | import csv |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 439 | with open('some.csv', 'w', newline='') as f: |
Eli Bendersky | 6860a92 | 2011-03-11 15:47:36 +0200 | [diff] [blame] | 440 | writer = csv.writer(f) |
| 441 | writer.writerows(someiterable) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 442 | |
R. David Murray | 8b7d4aa | 2009-04-04 01:38:38 +0000 | [diff] [blame] | 443 | Since :func:`open` is used to open a CSV file for reading, the file |
| 444 | will by default be decoded into unicode using the system default |
| 445 | encoding (see :func:`locale.getpreferredencoding`). To decode a file |
| 446 | using a different encoding, use the ``encoding`` argument of open:: |
| 447 | |
Eli Bendersky | 6860a92 | 2011-03-11 15:47:36 +0200 | [diff] [blame] | 448 | import csv |
| 449 | with open('some.csv', newline='', encoding='utf-8') as f: |
| 450 | reader = csv.reader(f) |
| 451 | for row in reader: |
| 452 | print(row) |
R. David Murray | 8b7d4aa | 2009-04-04 01:38:38 +0000 | [diff] [blame] | 453 | |
| 454 | The same applies to writing in something other than the system default |
| 455 | encoding: specify the encoding argument when opening the output file. |
| 456 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 457 | Registering a new dialect:: |
| 458 | |
| 459 | import csv |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 460 | csv.register_dialect('unixpwd', delimiter=':', quoting=csv.QUOTE_NONE) |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 461 | with open('passwd', newline='') as f: |
Eli Bendersky | 6860a92 | 2011-03-11 15:47:36 +0200 | [diff] [blame] | 462 | reader = csv.reader(f, 'unixpwd') |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 463 | |
| 464 | A slightly more advanced use of the reader --- catching and reporting errors:: |
| 465 | |
| 466 | import csv, sys |
Eli Bendersky | 6860a92 | 2011-03-11 15:47:36 +0200 | [diff] [blame] | 467 | filename = 'some.csv' |
| 468 | with open(filename, newline='') as f: |
| 469 | reader = csv.reader(f) |
| 470 | try: |
| 471 | for row in reader: |
| 472 | print(row) |
| 473 | except csv.Error as e: |
| 474 | sys.exit('file {}, line {}: {}'.format(filename, reader.line_num, e)) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 475 | |
| 476 | And while the module doesn't directly support parsing strings, it can easily be |
| 477 | done:: |
| 478 | |
| 479 | import csv |
| 480 | for row in csv.reader(['one,two,three']): |
Georg Brandl | 6911e3c | 2007-09-04 07:15:32 +0000 | [diff] [blame] | 481 | print(row) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 482 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 483 | |
R. David Murray | 8b7d4aa | 2009-04-04 01:38:38 +0000 | [diff] [blame] | 484 | .. rubric:: Footnotes |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 485 | |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 486 | .. [1] If ``newline=''`` is not specified, newlines embedded inside quoted fields |
| 487 | will not be interpreted correctly, and on platforms that use ``\r\n`` linendings |
R David Murray | 9c0d5ea | 2011-03-20 11:18:21 -0400 | [diff] [blame] | 488 | on write an extra ``\r`` will be added. It should always be safe to specify |
R David Murray | 9188702 | 2011-03-19 22:30:14 -0400 | [diff] [blame] | 489 | ``newline=''``, since the csv module does its own (universal) newline handling. |