如何允许用户根据用户输入在程序中存储多个CSV文件？- Python

import csv

filename = input("请输入文件名：")
data = input("请输入数据，以逗号分隔：")

data_list = data.split(",")

with open(filename, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(data_list)

while True:
    filename = input("请输入文件名（输入q退出）：")
    if filename == 'q':
        break
    data = input("请输入数据，以逗号分隔：")
    data_list = data.split(",")
    with open(filename, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(data_list)