python - matplotlib子图网格: where to insert row/column arguments

Question

我正在尝试以 matplotlib 子图的形式显示跨多个数据集的 LDA 文本分析的主题提取结果。

这是我现在的位置:

我认为我的问题是我对 matplotlib 不熟悉。我已经提前完成了所有的数字处理，以便我可以专注于如何绘制数据:

top_words_master = []
top_weights_master = []
for i in range(len(tf_list)):
    tf = tf_vectorizer.fit_transform(tf_list[i])
    lda.fit(tf)
    n_top_words = 20
    tf_feature_names = tf_vectorizer.get_feature_names_out()
    top_features_ind = lda.components_[0].argsort()[: -n_top_words - 1 : -1]
    top_features = [tf_feature_names[i] for i in top_features_ind]
    weights = lda.components_[0][top_features_ind]
    top_words_master.append(top_features)
    top_weights_master.append(weights)

这给了我我的单词和权重(x 轴值)来制作行/条形图的子图矩阵。

我尝试通过 matplot lib 构建它:

fig, axes = plt.subplots(2, 5, figsize=(30, 15), sharex=True)
plt.subplots_adjust(hspace=0.5)
fig.suptitle("Topics in LDA Model", fontsize=18, y=0.95)
axes = axes.flatten()
for i in range(len(tf_list)):

    ax = axes[i]
    ax.barh(top_words_master[i], top_weights_master[i], height=0.7)
    ax.set_title(topic_map[f"Topic {i +1}"], fontdict={"fontsize": 30})
    ax.invert_yaxis()
    ax.tick_params(axis="both", which="major", labelsize=20)
    for j in "top right left".split():
        ax.spines[j].set_visible(False)
    fig.suptitle("Topics in LDA Model", fontsize=40)

    plt.subplots_adjust(top=0.90, bottom=0.05, wspace=0.90, hspace=0.3)
    plt.show()

但是，它只显示了一个，第一个。对于剩下的 6 个数据集，它刚刚打印:

<Figure size 432x288 with 0 Axes> <Figure size 432x288 with 0 Axes> <Figure size 432x288 with 0 Axes> <Figure size 432x288 with 0 Axes> <Figure size 432x288 with 0 Axes>

问题

我已经为此好几天了。我觉得我已经很接近了，但是这种结果真的让我很困惑，有人有解决方案或能够指出我正确的方向吗？

Answer 1

据我从您的问题中了解到，您的问题是为子图获取正确的索引。

就您而言，您有一个数组 range(len(tf_list)) 来索引数据，一些数据(例如 top_words_master[i])要绘制，并且具有 10 个子图的图(行数 = 2，列数 = 5)。例如，如果您要绘制数据的第 7 个项目 (i=6)，则 ax 的索引将为 axes[1,1]。

为了获得子图轴的正确索引，您可以使用 numpy.unravel_index 。当然，您不应该压平您的轴。

import matplotlib.pyplot as plt
import numpy as np


# dummy function
my_func = lambda x: np.random.random(x)
x_max = 100

# fig properties
rows = 2
cols = 5
fig, axes = plt.subplots(rows,cols,figsize=(30, 15), sharex=True)

for i in range(rows*cols):
    ax_i = np.unravel_index(i,(rows,cols))
    
    axes[ax_i[0],ax_i[1]].barh(np.arange(x_max),my_func(x_max), height=0.7)

plt.show()