用python写一个程序，在n*n方格子内生成不同颜色，n分别取2，3，4，5

以下是一个使用Python的Tkinter模块生成不同颜色方格的示例程序，可以根据需要修改n的值来生成不同大小的方格：

import tkinter as tk
import random

def generate_color():
    """生成一个随机颜色"""
    r = lambda: random.randint(0, 255)
    return '#{:02X}{:02X}{:02X}'.format(r(), r(), r())

def create_grid(n):
    """在n*n方格内生成不同颜色的方格"""
    root = tk.Tk()
    for i in range(n):
        for j in range(n):
            color = generate_color()
            frame = tk.Frame(root, bg=color, width=50, height=50)
            frame.grid(row=i, column=j)
    root.mainloop()

if __name__ == '__main__':
    n = 5  # 修改n的值来生成不同大小的方格
    create_grid(n)

这个程序使用`generate_color()`函数来生成随机颜色，使用了lambda表达式和字符串格式化来生成RGB颜色代码。随机颜色生成后，程序会在n*n的Tkinter窗口中生成不同颜色的方格。