Python写入csv文件为每个像元分配一个字母

import csv
import string

def assign_letter():
    letters = string.ascii_uppercase  # 获取所有大写字母
    index = 0  # 初始化字母索引
    while True:
        yield letters[index]  # 生成下一个字母
        index = (index + 1) % len(letters)  # 循环使用字母

data = [[x, y] for x in range(10) for y in range(10)]  # 生成一个10x10的像元坐标列表

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    letter_generator = assign_letter()  # 创建字母生成器
    for x, y in data:
        letter = next(letter_generator)  # 获取下一个字母
        writer.writerow([x, y, letter])  # 写入数据