Python-从CSV文件读取数据

时间:2020-02-23 14:43:12  来源:igfitidea点击:

在本教程中,我们将学习在Python中读取CSV文件。

什么是CSV文件?

CSV或者逗号分隔值是纯文本文件,包含可打印ASCII或者Unicode字符的表格数据。

注意点!

CSV文件以.csv扩展名保存。

您可以使用记事本,Sublime Text等文本编辑器创建CSV文件。

您也可以使用MS-Excel之类的应用程序来创建CSV文件。

示例CSV文件

在以下示例中,我们具有一个示例CSV文件的内容,该文件包含4行和3列。

id,name,score
1,Jane Doe,10
2,John Doe,9
3,Alice,10

文件的第一行通常是列的名称。

上面的文件有3列-id,name和score。
其余三行包含数据。

导入csv模块

为了在Python中使用CSV文件,我们使用csv模块的帮助。

通过编写以下行来导入csv模块。

import csv

读取CSV文件

为了读取CSV文件数据,我们利用csv模块的reader()方法的帮助。

在下面的Python程序中,我们正在读取data.csv文件的内容并在控制台中打印其内容。

data.csv文件的内容。

id,name,score
1,Jane Doe,10
2,John Doe,9
3,Alice,10

读取CSV文件的Python代码。

# import module
import csv

try:
    # open file
    fobj = open('data.csv', 'r')
    
    # reader object
    csvReader = csv.reader(fobj)
    
    # get the field name
    fields = csvReader.next()
    
    # total columns
    columnsCount = len(fields)
    
    # iterate the rows
    rows = []
    for row in csvReader:
        rows.append(row)
    
    # print rows
    for r in rows:
        for i in range(columnsCount):
            print(fields[i] + " = " + r[i])
        print("------")
    
except:
    print("An error occurred while reading the file.")

finally:
    fobj.close()

上面的Python代码将为我们提供以下输出。

id = 1
name = Jane Doe
score = 10
-----
id = 2
name = John Doe
score = 9
-----
id = 3
name = Alice
score = 10
-----