使用 for 循环填充时，matplotlib 散点图不显示

Question

我正在尝试编写一个函数来使用 Python 3.4 中的 matplotlib 生成 Matlab 风格的相关图（例如 here）。但是，我想更改绘图，以便对角线子图显示变量的名称，下三角子图显示 Pearson 相关系数，上三角子图显示散点图。下面是一些生成示例数据的代码和我编写的函数。它会在正确的位置显示带有变量名称和相关系数的适当 4x4 子图网格，但散点图不显示。

import numpy as np
import matplotlib.pyplot as plt

means = [0, 1, 0, 2]
sig = [[1, 0.5, 0, -0.1], [0.5, 3, 0, 0.2], [0, -0.1, 1, -0.3], [-0.1, 0.2, -0.3, 1]]
data = np.random.multivariate_normal(means, sig, 50)
names = ['Var' + str(i) for i in range(data.shape[1])]

def corrplot(data, names):
    corrMat = np.corrcoef(data, rowvar = 0)
    numVars = data.shape[1]

    fig, ax = plt.subplots(numVars, numVars, sharex = "col", sharey = "row")
    fig.subplots_adjust(wspace = 0, hspace = 0)

    for i in range(numVars):
        for j in range(numVars):
            if i == j: # On the diagonal
                ax[i, j].text(0.5, 0.5, names[i], transform = ax[i, j].transAxes)
            elif i < j: # In the upper triangle
                ax[i, j].scatter(data[:, i], data[:, j], marker = '.')
            elif i > j: # In the lower triangle
                ax[i, j].text(0.5, 0.5, str(round(corrMat[i, j], 3)), transform = ax[i, j].transAxes)
    plt.show()

为了确定问题的根源，我使用以下代码手动重建了 2 变量情况的图，从而生成了所需的图：

fig, ax = plt.subplots(2, 2, sharex = "col", sharey = "row")
fig.subplots_adjust(wspace = 0, hspace = 0)
ax[0, 0].text(0.5, 0.5, 'Var0', transform = ax[0, 0].transAxes)
ax[0, 1].scatter(data[:, i], data[:, j], marker = '.')
ax[1, 0].text(0.5, 0.5, '0.5', transform = ax[1, 0].transAxes)
ax[1, 1].text(0.5, 0.5, 'Var1', transform = ax[1, 1].transAxes)
plt.show()

由于这行得通，我假设问题与在子图中混合文本和数据无关。我编写了下一个函数来测试使用 for 循环填充子图，它会按预期在每个子图中生成一个散点图。

def test1(data):
    numVars = data.shape[1]
    fig, ax = plt.subplots(numVars, numVars, sharex = "col", sharey = "row")
    fig.subplots_adjust(wspace = 0, hspace = 0)

    for i in range(numVars):
        for j in range(numVars):
            ax[i, j].scatter(data[:, i], data[:, j], marker = '.')
    plt.show()

接下来，我尝试使用 for 循环仅填充子图的一个子集。这会产生一个空白网格，如下所示。

def test2(data):
    numVars = data.shape[1]
    fig, ax = plt.subplots(numVars, numVars, sharex = "col", sharey = "row")
    fig.subplots_adjust(wspace = 0, hspace = 0)

    for i in range(numVars):
        for j in range(i + 1, numVars):
            ax[i, j].scatter(data[:, i], data[:, j], marker = '.')
    plt.show()

这让我相信存在一些与 for 循环以及如何创建散点图有关的错误，但我还没有找到错误。

Answer 1

您的代码准确地显示了所需的情节。 我认为您的matplolib 版本无法识别marker = '.'

您可以尝试使用默认标记（不带marker = '.'）进行绘图或将其替换为marker = 'o'