如何在Python中通过字符串调用Seaborn不同绘图函数？

使用Seaborn生成直方图的代码如下：

ax = sns.histplot(data=data, y=value, x=category)
ax.figure.set_size_inches(7, len(10))  # 注：此处len(10)会报错，int类型无len属性，应改为具体数值如10
plt.title('Title'.format(field))
plt.show()

想要批量生成直方图、箱线图、小提琴图，定义了绘图类型字符串列表：

plot_type_list = ['histplot', 'boxplot', 'violinplot']

尝试循环调用对应函数时，代码如下：

for plot_type in plot_type_list:
    ax = sns.plot_type(data=data, y=value, x=category)
    ax.figure.set_size_inches(7, len(10))
    plt.title('Title'.format(field))
    plt.show()

运行后报错：

AttributeError: module 'seaborn' has no attribute 'plot_type'

问题原因：Python将plot_type当作seaborn模块的固定属性名，而非变量替换为列表中的字符串，无法正确匹配到目标绘图函数。

有两种可行的修改方式：

方法1：使用getattr()获取函数 #

通过getattr()函数，根据字符串从seaborn模块中获取对应的绘图函数对象，再调用该函数：

import seaborn as sns
import matplotlib.pyplot as plt

plot_type_list = ['histplot', 'boxplot', 'violinplot']

for plot_type in plot_type_list:
    # 根据字符串获取对应的Seaborn绘图函数
    plot_func = getattr(sns, plot_type)
    ax = plot_func(data=data, y=value, x=category)
    ax.figure.set_size_inches(7, 10)  # 修正原代码中len(10)的错误
    plt.title(f'Title for {plot_type}')  # 可选：动态修改标题区分不同图表类型
    plt.show()

方法2：直接存储函数对象到列表 #

跳过字符串匹配步骤，直接将Seaborn的绘图函数对象存入列表，循环时直接调用：

import seaborn as sns
import matplotlib.pyplot as plt

# 直接存储函数对象，而非字符串
plot_type_list = [sns.histplot, sns.boxplot, sns.violinplot]

for plot_func in plot_type_list:
    ax = plot_func(data=data, y=value, x=category)
    ax.figure.set_size_inches(7, 10)
    plt.title(f'Title for {plot_func.__name__}')  # 通过__name__属性获取函数名
    plt.show()

额外说明 #

原代码中的len(10)会触发TypeError: object of type 'int' has no len()错误，需改为具体的数值（如10）来设置图表高度。

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