Python中os模块、csv模块和xlrd模块的使用

1. os模块的使用

1.1 获取当前工作目录

在Python中，我们可以使用os模块来进行操作系统相关的功能。其中一个常见的功能是获取当前的工作目录，即当前脚本所在的目录。

import os

current_dir = os.getcwd()
print(f"当前工作目录：{current_dir}")

在上述代码中，我们使用了os模块的getcwd()函数来获取当前工作目录，并将结果打印出来。

重要提示：在使用os模块的函数时，需要先导入os模块。

1.2 切换工作目录

除了获取当前工作目录外，我们还可以使用os模块来切换工作目录。这对于需要在不同目录下进行文件操作的情况非常有用。

import os

# 切换到指定目录
os.chdir('/path/to/directory')
# 切换回原来的工作目录
os.chdir(current_dir)

在上述代码中，我们使用了os模块的chdir()函数来切换工作目录。首先，我们将工作目录切换到指定的目录，然后又切换回原来的工作目录。

2. csv模块的使用

2.1 读取CSV文件

CSV（Comma-Separated Values）文件是一种常见的文件格式，用于存储表格数据。Python的csv模块提供了读取和写入CSV文件的功能。

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    reader = csv.reader(file)
    
    # 读取并打印每一行数据
    for row in reader:
        print(row)

在上述代码中，我们首先使用open()函数打开CSV文件，并指定模式为'r'，表示只读模式。然后，创建了一个csv.reader对象，用于读取CSV文件。最后，使用for循环逐行读取并打印每一行数据。

2.2 写入CSV文件

除了读取CSV文件外，csv模块还可以用于写入CSV文件。

import csv

# 打开CSV文件
with open('data.csv', 'w') as file:
    # 创建CSV写入器
    writer = csv.writer(file)
    
    # 写入数据
    writer.writerow(['Name', 'Age', 'Gender'])
    writer.writerow(['John', '25', 'Male'])
    writer.writerow(['Alice', '30', 'Female'])

在上述代码中，我们首先使用open()函数打开CSV文件，并指定模式为'w'，表示写入模式。然后，创建了一个csv.writer对象，用于写入CSV文件。接下来，使用writerow()函数来逐行写入数据。

3. xlrd模块的使用

3.1 读取Excel文件

除了CSV文件外，Python的xlrd模块还可以用于读取Excel文件。

import xlrd

# 打开Excel文件
workbook = xlrd.open_workbook('data.xlsx')
# 获取第一个表格
sheet = workbook.sheet_by_index(0)
# 读取并打印每一行数据
for row_idx in range(sheet.nrows):
    row_data = sheet.row_values(row_idx)
    print(row_data)

在上述代码中，我们首先使用open_workbook()函数打开Excel文件，并获取工作簿workbook。然后，通过sheet_by_index()函数获取第一个表格sheet。接下来，使用row_values()函数逐行读取并打印每一行数据。

3.2 写入Excel文件

除了读取Excel文件外，xlrd模块还可以用于写入Excel文件。

import xlwt

# 创建Excel工作簿
workbook = xlwt.Workbook()
# 创建第一个表格
sheet = workbook.add_sheet('Sheet1')
# 写入数据
sheet.write(0, 0, 'Name')
sheet.write(0, 1, 'Age')
sheet.write(1, 0, 'John')
sheet.write(1, 1, '25')
sheet.write(2, 0, 'Alice')
sheet.write(2, 1, '30')
# 保存Excel文件
workbook.save('data.xls')

在上述代码中，我们首先使用Workbook()函数创建Excel工作簿workbook。然后，使用add_sheet()函数创建第一个表格sheet。接下来，使用write()函数来逐个单元格写入数据。最后，使用save()函数保存Excel文件。

总结：

在Python中，os模块、csv模块和xlrd模块是常用的模块，用于进行操作系统相关的功能、读取和写入CSV文件、读取和写入Excel文件。学会使用这些模块，可以帮助我们更方便地进行文件操作和数据处理。