如何将绘制的图形自动保存为源文件名？

import matplotlib.pyplot as plt

def save_plot_as_source_file_name(source_file_name):
    # 绘制图形
    plt.plot([1, 2, 3, 4], [1, 4, 9, 16])
    
    # 保存图形为源文件名
    plt.savefig(source_file_name)
    
    # 关闭图形窗口
    plt.close()

# 获取源文件名，这里假设源文件名为"example.png"
source_file_name = "example.png"

# 调用函数保存图形
save_plot_as_source_file_name(source_file_name)