我有一个模型,它将XY平面上的点分成两部分:它返回每个输入对的值范围(0, -1)
,在橙色上方和蓝线下方,在它返回(0, +1)
的线之间。如何根据这些值填充颜色渐变区域?这应该是很常见的事情,但我没有找到比fill_between
函数更有用的函数。
发布于 2020-01-31 00:04:07
您可以根据您的模型值(在下面的代码中名为z
)创建一个2D数组。然后使用该z值对像素进行着色,确保对称地设置vmin
和vmax
,以使中心颜色标记z
的零。如果颜色太亮,适当的alpha可以帮助柔化它们。
from matplotlib import pyplot as plt
import numpy as np
N = 20
red_dots_x = np.concatenate([np.random.normal(0.5, 0.1, N), np.random.normal(-0.5, 0.1, N)])
red_dots_y = np.concatenate([np.random.normal(-0.5, 0.1, N), np.random.normal(0.5, 0.1, N)])
blue_dots_x = np.concatenate([np.random.normal(0.5, 0.1, N), np.random.normal(-0.5, 0.1, N)])
blue_dots_y = np.concatenate([np.random.normal(0.5, 0.1, N), np.random.normal(-0.5, 0.1, N)])
plt.plot(red_dots_x, red_dots_y, 'ro', ls='')
plt.plot(blue_dots_x, blue_dots_y, 'b*', ls='')
x0, x1 = plt.xlim()
y0, y1 = plt.ylim()
x, y = np.meshgrid(np.linspace(x0, x1, 10), np.linspace(y0, y1, 10))
z = abs(x - y) - 0.6
plt.imshow(z, cmap='bwr', vmin=-1, vmax=1, interpolation='bilinear', alpha=0.4,
extent=[x0, x1, y0, y1], origin='lower')
plt.show()
https://stackoverflow.com/questions/59989184
复制相似问题