6.4 KiB

Raw Permalink Blame History

在 Python 中使用 CSV 模块

原文：https://www.pythonforbeginners.com/csv/using-the-csv-module-in-python

如果要导入或导出电子表格和数据库以便在 Python 解释器中使用，则必须依赖 CSV 模块或逗号分隔值格式。

什么是 CSV 文件？

CSV 文件用于存储大量变量或数据。它们是极其简化的电子表格——想想 Excel——只是内容以明文形式存储。

CSV 模块是一个内置函数，允许 Python 解析这些类型的文件。

值得注意的是，当您处理 CSV 文件时，您正在涉足 JSON 开发。

JSON——代表 JavaScript 对象符号——是一种格式，用于在明文文件中以 JavaScript 代码的形式存储信息。您不需要了解 JavaScript 来处理这些文件，实践也不局限于该语言。显然，因为我们在这里使用的是 Python。

CSV 文件中的文本按行排列，每一行都有列，用逗号分隔。文件中的每一行都是电子表格中的一行，而逗号用于定义和分隔单元格。

使用 CSV 模块

要从 CSV 文件中提取信息，可以使用循环和分割方法从各个列中获取数据。

CSV 模块的存在就是为了处理这个任务，使得处理 CSV 格式的文件变得更加容易。当您处理从实际的电子表格和数据库导出到文本文件的数据时，这变得尤其重要。这些信息本身很难读懂。

不幸的是，没有标准，所以 CSV 模块使用“方言”来支持使用不同参数的解析。除了通用读取器和写入器，该模块还包括一种用于处理 Microsoft Excel 和相关文件的方言。

CSV 功能

CSV 模块包括所有内置的必要功能。它们是:

csv .阅读器
csv.writer
csv.register_dialect
csv.unregister_dialect
csv.get_dialect
csv.list _ 方言
csv.field_size_limit

在本指南中，我们将只关注 reader 和 writer 函数，它们允许您编辑、修改和操作 CSV 文件中存储的数据。

读取 CSV 文件

要从 CSV 文件中提取数据，必须使用 reader 函数来生成 reader 对象。

reader 函数被设计成获取文件的每一行并列出所有列。然后，您只需选择需要可变数据的列。

听起来比实际复杂多了。为了证明这一点，我们来看一个例子。

*i**mport CSV*
*With* *open(**‘some.csv’, ‘**rb**’) as f:*
*r**eader =* *csv.reader**(f)*
*f**or row in reader:*
*p**rint row*

注意第一个命令是如何用于导入 CSV 模块的？

让我们看另一个例子。

*i**mport csv* 
*i**mport sys*

*f* *= open(**sys.argv**[1], ‘**rb**’)*
*r**eader =* *csv.reader**(f)*
*f**or row in reader*
*p**rint row*

*f.close**()*

在前两行中，我们导入 CSV 和 sys 模块。然后，我们打开想要从中提取信息的 CSV 文件。

接下来，我们创建 reader 对象，迭代文件的行，然后打印它们。最后，我们结束操作。

CSV 样本文件

我们将看看一个示例 CSV 文件。注意信息是如何存储和呈现的。

*Title,Release* *Date,Director*
*And Now For Something Completely Different,**1971,Ian* *MacNaughton*
*Monty Python And The Holy Grail,**1975,Terry* *Gilliam and Terry Jones*
*Monty Python's Life Of Brian,**1979,Terry* *Jones*
*Monty Python Live At The Hollywood Bowl,**1982,Terry* *Hughes*
*Monty Python's The Meaning Of Life,**1983,Terry* *Jones*

读取 CSV 文件示例

我们将从一个基本的 CSV 文件开始，它有 3 列，包含变量“A”、“B”、“C”和“D”。

*$ cat test.csv*
*A,B**,”C D”*
*1,2,”3 4”*
*5,6,7*

然后，我们将使用下面的 Python 程序来读取和显示上述 CSV 文件的内容。

*import csv*

*ifile* *=* *open(**‘test.csv’, “**rb**”)*
*reader =* *csv.reader**(**ifile**)*

*rownum* *= 0*
*for row in reader:*
*# Save header row.*
*i**f* *rownum* *==0:*
*h**eader = row*
*e**lse:*
*c**olnum* *= 0*
*f**or col in row:*
*p**rint ‘%-8s: %s’ % (header[**colnum**], col)*
*c**olnum* *+ = 1*

*r**ownum* *+ = 1*

*i**file.close**()*

当我们用 Python 执行这个程序时，输出将如下所示:

*$ python csv1.py*
*A    **  :* *1* 
*B    **  :* *2*
*C D    **  :* *3 4*
*A    **  :** 5* 
*B    **  :* *6*
*C D    **  :* *7*

写入 CSV 文件

当您有一组想要存储在 CSV 文件中的数据时，是时候反过来使用 write 函数了。信不信由你，这和阅读它们一样容易做到。

writer() 函数将创建一个适合写的对象。要迭代行上的数据，需要使用writerow(*)***函数。

这里有一个例子。

以下 Python 程序将名为“test.csv”的文件转换为 csv 文件，该文件使用制表符作为值分隔符，所有值都用引号括起来。分隔符和引号，以及如何/何时引用，是在创建编写器时指定的。创建 reader 对象时，这些选项也是可用的。

*import csv*

*ifile**  =* *open('test.csv', "**rb**")*
*reader =* *csv.reader**(**ifile**)*
*ofile**  =* *open('ttest.csv', "**wb**")*
*writer =* *csv.writer**(**ofile**, delimiter='**',* *quotechar**='"', quoting=**csv.QUOTE_ALL**)*

*for row in reader:*
*writer.writerow**(row)*

*ifile.close**()*
*ofile.close**()*

当您执行这个程序时，输出将是:

*$ python csv2.py*
*$ cat ttest.csv*
*"A"     "B"     "C D"*
*"1"     "2"     "3 4"*
*"5"     "6"     "7"*

引用 CSV 文件

使用 CSV 模块，您还可以执行各种报价功能。

它们是:

csv。引用_ 所有**–引用所有内容，不考虑类型。
csv。引用_ 最小**–引用包含特殊字符的字段
csv。引用_ 非数字**–引用所有非整数或浮点数的字段
csv。引用_ 无**–不引用输出上的任何内容

6.4 KiB Raw Permalink Blame History Unescape Escape