编写Python函数来处理CSV和Excel文件

CSV和Excel是两种常见的电子表格文件格式，用于存储和管理数据。Python提供了强大的库和模块来处理和操作这些文件。

CSV（逗号分隔值）是一种简单的文本文件格式，用于存储表格数据。每行都是一条记录，并用逗号将不同的字段分隔开来。使用Python处理CSV文件需要使用csv模块。

Excel是一种流行的电子表格文件格式，它支持多个工作表和更复杂的数据结构。在Python中处理Excel文件需要使用第三方库openpyxl或者pandas。

接下来，我们将详细介绍如何使用Python处理CSV和Excel文件。

处理CSV文件：

1. 导入csv模块：首先，我们需要导入csv模块以便使用其提供的函数来处理CSV文件。

   import csv
   

2. 读取CSV文件：使用csv模块的reader函数可以方便地读取CSV文件中的内容。

   with open('data.csv', 'r') as file:
       csv_reader = csv.reader(file)
       for row in csv_reader:
           print(row)
   

上述代码中，我们打开一个名为data.csv的文件，并使用csv.reader函数创建一个CSV Reader对象。然后，使用for循环遍历CSV文件中的每一行，并将其打印出来。

3. 写入CSV文件：使用csv模块的writer函数可以将数据写入CSV文件。

   with open('data.csv', 'w') as file:
       csv_writer = csv.writer(file)
       csv_writer.writerow(['Name', 'Age', 'Country'])
       csv_writer.writerow(['John', '25', 'USA'])
   

上述代码中，我们打开一个名为data.csv的文件，并使用csv.writer函数创建一个CSV Writer对象。然后，使用writerow函数将一行数据写入CSV文件。在上述例子中，我们将标题行和数据行写入CSV文件。

处理Excel文件：

1. 导入库：

   import openpyxl
   import pandas as pd
   

2. 读取Excel文件：

- 使用openpyxl库：

     workbook = openpyxl.load_workbook('data.xlsx')
     sheet = workbook.active
     for row in sheet.iter_rows():
         for cell in row:
             print(cell.value)
     

使用openpyxl库，我们可以打开Excel文件并访问其中的工作表。使用iter_rows方法，我们可以遍历工作表中的每一行，并使用cell.value来获取单元格的值。

- 使用pandas库：

     df = pd.read_excel('data.xlsx')
     print(df)
     

使用pandas库读取Excel文件更加简单，我们可以使用read_excel函数直接加载整个Excel文件，并将其转换为DataFrame对象。

3. 写入Excel文件：

- 使用openpyxl库：

     workbook = openpyxl.Workbook()
     sheet = workbook.active
     sheet['A1'] = 'Name'
     sheet['B1'] = 'Age'
     sheet['C1'] = 'Country'
     sheet['A2'] = 'John'
     sheet['B2'] = 25
     sheet['C2'] = 'USA'
     workbook.save('data.xlsx')
     

使用openpyxl库，我们可以创建一个新的工作簿，并在其中创建工作表。然后，我们可以使用工作表的单元格索引来设置单元格的值，最后使用save方法保存工作簿。

- 使用pandas库：

     df = pd.DataFrame({'Name': ['John'], 'Age': [25], 'Country': ['USA']})
     df.to_excel('data.xlsx', index=False)
     

使用pandas库，我们可以使用DataFrame对象来创建包含数据的表格。然后，我们可以使用to_excel函数将DataFrame对象保存为Excel文件。

以上是一些常见的处理CSV和Excel文件的方法，Python提供了许多其他功能，可以根据需要进行使用和定制化。