如何将包含键的字典中的数据保存为csv文件，并将包含数据的列表保存到各个键中

import csv

data_dict = {
    'key1': [1, 2, 3],
    'key2': [4, 5, 6],
    'key3': [7, 8, 9]
}

csv_file = 'data.csv'

with open(csv_file, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(data_dict.keys())
    writer.writerows(zip(*data_dict.values()))

for key, values in data_dict.items():
    data_dict[key] = values

import csv

data_dict = {
    'key1': [1, 2, 3],
    'key2': [4, 5, 6],
    'key3': [7, 8, 9]
}

csv_file = 'data.csv'

with open(csv_file, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(data_dict.keys())
    writer.writerows(zip(*data_dict.values()))

for key, values in data_dict.items():
    data_dict[key] = values