是否使用python删除csv文件中的数据？

import csv
import os

def delete_data(csv_file, row_index):
    # 创建一个临时文件
    temp_file = csv_file + '.temp'
    
    # 打开原始CSV文件和临时文件
    with open(csv_file, 'r') as file, open(temp_file, 'w', newline='') as temp:
        reader = csv.reader(file)
        writer = csv.writer(temp)
        
        # 遍历原始文件的每一行
        for i, row in enumerate(reader):
            # 如果行索引不等于要删除的行索引，则将该行写入临时文件
            if i != row_index:
                writer.writerow(row)
    
    # 关闭文件
    file.close()
    temp.close()
    
    # 删除原始文件
    os.remove(csv_file)
    
    # 将临时文件重命名为原始文件名
    os.rename(temp_file, csv_file)

# 调用函数删除第3行数据（索引从0开始）
delete_data('data.csv', 2)