如何在matplotlib绘图的xlabel中打印10K，20K...1M

import matplotlib.pyplot as plt
import matplotlib.ticker as ticker

# 创建图表和子图
fig, ax = plt.subplots()

# 绘制数据
x = [i for i in range(10)]
y = [i*1000 for i in range(10)]
ax.plot(x, y)

# 设置x轴刻度范围
ax.set_xlim(0, 9)

def format_func(value, tick_number):
    # 选择单位和倍数
    units = ['K', 'M']
    multiplier = 1000
    
    # 计算指数
    exponent = int(tick_number / 3)
    # 计算单位索引
    unit_index = min(exponent - 1, len(units) - 1)
    
    # 将数值转换为字符串
    value = value / (multiplier ** exponent)
    value_str = f'{value:.0f}{units[unit_index]}'
    
    return value_str

# 创建x轴刻度格式化器
formatter = ticker.FuncFormatter(format_func)

# 设置x轴刻度格式化器
ax.xaxis.set_major_formatter(formatter)

# 显示图表
plt.show()