matplotlib轴上的千(K)和兆(M)后缀

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

def kilo_mega_formatter(x, pos):
    if x >= 1e6:
        return f"{x/1e6:.0f}M"
    elif x >= 1e3:
        return f"{x/1e3:.0f}K"
    else:
        return f"{x:.0f}"

fig, ax = plt.subplots()

# 设置x轴和y轴的单位换算
ax.xaxis.set_major_formatter(ticker.FuncFormatter(kilo_mega_formatter))
ax.yaxis.set_major_formatter(ticker.FuncFormatter(kilo_mega_formatter))

# 绘制图形
ax.plot([1000, 10000, 100000], [1000, 10000, 100000])

# 显示图形
plt.show()