在Python中读取和写入 CSV 文件

CSV（逗号分隔值）格式是电子表格和数据库最常见的导入和导出格式。它是在应用程序和数据科学中使用的流行数据格式之间交换数据的最常用方法之一。它受到广泛的应用程序的支持。 CSV 文件存储表格数据，其中每个数据字段由分隔符（大多数情况下为逗号）分隔。要表示 CSV 文件，它必须以.csv文件扩展名保存。

从 CSV 文件中读取

Python包含一个名为csv的模块，用于处理 CSV 文件。模块中的 reader 类用于从 CSV 文件中读取数据。首先，CSV 文件在 'r' 模式下使用 open() 方法打开（指定打开文件时的读取模式），该方法返回文件对象，然后使用 CSV 模块的reader()方法读取它，该方法返回遍历指定 CSV 文档中各行的 reader 对象。

句法：

csv.reader(csvfile, dialect='excel', **fmtparams

注意： 'with' 关键字与open()方法一起使用，因为它简化了异常处理并自动关闭 CSV 文件。

例子：

考虑下面的 CSV 文件——

import csv 
    
# opening the CSV file 
with open('Giants.csv', mode ='r')as file: 
      
  # reading the CSV file 
  csvFile = csv.reader(file) 
    
  # displaying the contents of the CSV file 
  for lines in csvFile: 
        print(lines)

输出：

[['Steve', 13, 'A'],
['John', 14, 'F'],
['Nancy', 14, 'C'],
['Ravi', 13, 'B']]

写入 CSV 文件

csv.writer类用于将数据插入 CSV 文件。此类返回一个 writer 对象，该对象负责将用户的数据转换为分隔字符串。 CSV 文件对象应使用 newline=” 打开，否则，引用字段内的字符将无法正确解释。

句法：

csv.writer(csvfile, dialect='excel', **fmtparams)

csv.writer 类提供了两种写入 CSV 的方法。它们是writerow()和writerows() 。

writerow()：此方法一次写入一行。可以使用此方法写入字段行。
句法：
```
writerow(fields)
```
writerows()：此方法用于一次写入多行。这可用于编写行列表。
句法：
```
writerows(rows)
```

例子：

# Python program to demonstrate
# writing to CSV
  
  
import csv 
    
# field names 
fields = ['Name', 'Branch', 'Year', 'CGPA'] 
    
# data rows of csv file 
rows = [ ['Nikhil', 'COE', '2', '9.0'], 
         ['Sanchit', 'COE', '2', '9.1'], 
         ['Aditya', 'IT', '2', '9.3'], 
         ['Sagar', 'SE', '1', '9.5'], 
         ['Prateek', 'MCE', '3', '7.8'], 
         ['Sahil', 'EP', '2', '9.1']] 
    
# name of csv file 
filename = "university_records.csv"
    
# writing to csv file 
with open(filename, 'w') as csvfile: 
    # creating a csv writer object 
    csvwriter = csv.writer(csvfile) 
        
    # writing the fields 
    csvwriter.writerow(fields) 
        
    # writing the data rows 
    csvwriter.writerows(rows)

输出：

python-write-to-csv

我们还可以将字典写入 CSV 文件。为此，CSV 模块提供了 csv.DictWriter 类。此类返回一个将字典映射到输出行的写入器对象。

句法：

csv.DictWriter(csvfile, fieldnames, restval=”, extrasaction=’raise’, dialect=’excel’, *args, **kwds)

编程需要懂一点英语

csv.DictWriter 提供了两种写入 CSV 的方法。他们是：

writeheader()： writeheader()方法只是使用预先指定的字段名写入 csv 文件的第一行。
句法：
```
writeheader()
```
writerows()： writerows方法简单地写入所有行，但在每一行中，它只写入值（而不是键）。
句法：
```
writerows(mydict)
```

例子：

# importing the csv module 
import csv 
    
# my data rows as dictionary objects 
mydict =[{'branch': 'COE', 'cgpa': '9.0', 'name': 'Nikhil', 'year': '2'}, 
         {'branch': 'COE', 'cgpa': '9.1', 'name': 'Sanchit', 'year': '2'}, 
         {'branch': 'IT', 'cgpa': '9.3', 'name': 'Aditya', 'year': '2'}, 
         {'branch': 'SE', 'cgpa': '9.5', 'name': 'Sagar', 'year': '1'}, 
         {'branch': 'MCE', 'cgpa': '7.8', 'name': 'Prateek', 'year': '3'}, 
         {'branch': 'EP', 'cgpa': '9.1', 'name': 'Sahil', 'year': '2'}] 
    
# field names 
fields = ['name', 'branch', 'year', 'cgpa'] 
    
# name of csv file 
filename = "university_records.csv"
    
# writing to csv file 
with open(filename, 'w') as csvfile: 
    # creating a csv dict writer object 
    writer = csv.DictWriter(csvfile, fieldnames = fields) 
        
    # writing headers (field names) 
    writer.writeheader() 
        
    # writing data rows 
    writer.writerows(mydict)

输出：

python-csv

在评论中写代码？请使用 ide.geeksforgeeks.org，生成链接并在此处分享链接。