作为库使用时，如何在locust中输出csv？

import csv
from locust import HttpUser, task, between

class MyUser(HttpUser):
    wait_time = between(1, 2)

    @task
    def my_task(self):
        # 执行任务的代码

        # 输出数据到csv文件
        with open('output.csv', 'w', newline='') as file:
            writer = csv.writer(file)
            writer.writerow(['列1', '列2', '列3'])  # 写入表头
            writer.writerow(['数据1', '数据2', '数据3'])  # 写入数据

        # 其他任务代码

        file.close()