如何在Python中批量导入matplotlib等库以简化函数代码？

你之前的方法行不通，核心原因是函数内部导入的库仅存在于函数的局部命名空间，函数执行完后外部命名空间根本访问不到这些导入的对象。下面给你几种实用的解决思路：

方案1：用独立模块统一管理导入（最推荐） #

创建一个专门的导入管理模块，比如命名为plot_deps.py，把所有绘图需要的库都在这里导入：

# plot_deps.py
import matplotlib
import matplotlib.pyplot as plt
# 其他你需要的库，比如numpy、seaborn等
import numpy as np
import seaborn as sns

之后在任何需要绘图的脚本/函数文件里，直接从这个模块导入需要的对象即可：

# plot1.py
from plot_deps import matplotlib, plt, np, sns

def draw_line_chart(x_data, y_data):
    plt.figure(figsize=(10,6))
    plt.plot(x_data, y_data, color='blue', linewidth=2)
    plt.xlabel('X轴')
    plt.ylabel('Y轴')
    plt.show()

这种方式最简洁，也符合Python的模块导入规范，维护起来也方便——要加新库只需要修改plot_deps.py。

方案2：让导入函数返回需要的库对象 #

如果一定要用函数的方式实现，可以让函数把导入的库对象作为返回值，在调用时赋值到当前命名空间：

# script1.py
def load_plot_libs():
    import matplotlib
    import matplotlib.pyplot as plt
    import numpy as np
    return matplotlib, plt, np

然后在绘图脚本里：

# plot1.py
from script1 import load_plot_libs

# 调用函数获取库对象，赋值到当前命名空间
matplotlib, plt, np = load_plot_libs()

def draw_scatter_plot(x, y):
    plt.scatter(x, y, c='red', marker='o')
    plt.title('散点图示例')
    plt.show()

这种方式虽然能实现需求，但每次使用都要手动调用函数并赋值，不如方案1高效。

方案3：在绘图包的__init__.py中统一导入 #

如果你的绘图函数是一个Python包（比如有多个模块组成的库），可以把导入语句放在包的__init__.py里：

# my_plot_lib/__init__.py
import matplotlib
import matplotlib.pyplot as plt
import numpy as np

# 可选：指定对外暴露的对象，方便用from my_plot_lib import *导入
__all__ = ['matplotlib', 'plt', 'np']

之后在包内的各个模块中，直接从包本身导入：

# my_plot_lib/line_plots.py
from . import plt, np

def draw_multi_line(data_dict):
    plt.figure(figsize=(12,8))
    for label, data in data_dict.items():
        plt.plot(data['x'], data['y'], label=label)
    plt.legend()
    plt.show()

外部用户使用你的包时，也可以直接从包导入这些库对象，不用自己单独导入matplotlib。

内容的提问来源于stack exchange，提问作者Miss_Orchid