python - 使 matplotlib 等高线图给出填充的颜色条

Question

以 matplotlib 绘图为例

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 4, 1000)
y = np.linspace(-1, 1, 1000)
xx, yy = np.meshgrid(x, y)
z = np.sin(xx + yy**2)
plt.figure()
CS = plt.contourf(x, y, z)
CS = plt.contour(x, y, z)
plt.clabel(CS, fontsize=8, colors='black')
cbar = plt.colorbar(CS)
plt.savefig('test.png')

这会产生

颜色条很难阅读，因为它只放置颜色线，其余的都是白色的。

如何使颜色栏填充颜色，而不仅仅是线条，如 examples 中所示？

Answer 1

事实证明，无论您使用 contour 中的 CS 还是 contourf 中的 CS 都很重要。您希望使用 contourf 中的颜色条来获得填充的颜色条。

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 4, 1000)
y = np.linspace(-1, 1, 1000)
xx, yy = np.meshgrid(x, y)
z = np.sin(xx + yy**2)
plt.figure()
CS = plt.contour(x, y, z)
CS = plt.contourf(x, y, z)
plt.clabel(CS, fontsize=8, colors='black')
cbar = plt.colorbar(CS)
plt.savefig('test.png')

请注意我如何交换 contour 和 contourf 线。

我不知道如何仅使用轮廓制作填充颜色条。