AES（Advanced Encryption Standard）是一种对称加密算法，常用的密钥长度有128位、192位和256位。

在AES中，密钥扩展是指根据输入的密钥生成一系列的轮密钥，用于每一轮的加密操作。密钥扩展算法主要涉及字节代换（SubBytes）、行移位（ShiftRows）、列混淆（MixColumns）和轮密钥加（AddRoundKey）等步骤。

不同密钥长度对应着不同的轮数和轮密钥数量，因此，AES128、AES192和AES256在密钥扩展和字节生成方面有以下区别：

1. AES128：使用128位的密钥，共进行10轮加密操作，生成11个轮密钥。

2. AES192：使用192位的密钥，共进行12轮加密操作，生成13个轮密钥。

3. AES256：使用256位的密钥，共进行14轮加密操作，生成15个轮密钥。

下面是使用Python中的pycryptodome库来进行AES加密的示例代码：

from Crypto.Cipher import AES

def encrypt_aes128(key, plaintext):
    cipher = AES.new(key, AES.MODE_ECB)
    ciphertext = cipher.encrypt(plaintext)
    return ciphertext

def encrypt_aes192(key, plaintext):
    cipher = AES.new(key, AES.MODE_ECB)
    ciphertext = cipher.encrypt(plaintext)
    return ciphertext

def encrypt_aes256(key, plaintext):
    cipher = AES.new(key, AES.MODE_ECB)
    ciphertext = cipher.encrypt(plaintext)
    return ciphertext

# 示例使用 ECB 模式，实际使用时应该使用更安全的模式（如 CBC、CTR 等）和填充方式（如 PKCS7）

需要注意的是，以上示例代码只包含了加密部分，解密部分可以使用相同的密钥和相反的操作进行解密。同时，实际使用时应该采用更安全的模式和填充方式，以提供更好的安全性。