python - 在 matplotlib 中同时显示 2 个图，而不是一个接一个地显示

Question

我有以下代码，它首先显示了一个 matplotlib 图。然后，我必须关闭第一个图，以便出现第二个图。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import mglearn

# generate dataset
X, y = mglearn.datasets.make_forge()
# plot dataset
mglearn.discrete_scatter(X[:, 0], X[:, 1], y)
plt.legend(["Class 0", "Class 1"], loc=4)
plt.xlabel("First feature")
plt.ylabel("Second feature")
print("X.shape: {}".format(X.shape))

plt.show()

X, y = mglearn.datasets.make_wave(n_samples=40)
plt.plot(X, y, 'o')
plt.ylim(-3, 3)
plt.xlabel("Feature")
plt.ylabel("Target")

plt.show()

我想让 2 个 matplotlib 图同时出现。

Answer 1

plt.show() 绘制状态机中存在的所有图形。仅在脚本末尾调用它，确保绘制所有先前创建的图形。

现在您需要确保每个绘图确实是在不同的图形中创建的。这可以使用 plt.figure(fignumber) 实现，其中 fignumber 是从索引 1 开始的数字。

import matplotlib.pyplot as plt
import mglearn

# generate dataset
X, y = mglearn.datasets.make_forge()

plt.figure(1)
mglearn.discrete_scatter(X[:, 0], X[:, 1], y)
plt.legend(["Class 0", "Class 1"], loc=4)
plt.xlabel("First feature")
plt.ylabel("Second feature")


plt.figure(2)
X, y = mglearn.datasets.make_wave(n_samples=40)
plt.plot(X, y, 'o')
plt.ylim(-3, 3)
plt.xlabel("Feature")
plt.ylabel("Target")

plt.show()