使用imshow绘制hsv值的图像

Question

我正在尝试使用matplotlib中的imshow绘制hsv值。问题在于我使用的方法返回了一个包含三个值的元组，符合hsv的预期结果，但是imshow却将其解释为rgb。有没有办法告诉imshow这些值是hsv值呢？

以下是我的代码：

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as mcolors


def G(x, y):
    s = x + 1j*y
    return (s+2)/(s**2 + s + 1)

x = np.linspace(-3, 3, 1000)
y = np.linspace(-3, 3, 1000)

xx, yy = np.meshgrid(x, y)
norm = mcolors.Normalize()
zz = G(xx, yy)
phase = np.angle(zz)
mag = np.abs(zz)

# color converter
c = mcolors.ColorConverter().to_rgb

# Custom rgb Colormap
rgb = make_colormap(
    [c('red'), c('yellow'), 0.33, c('yellow'), c('green'), c('cyan'), 0.5, c('cyan'),
     c('blue'), c('magenta'), 0.833, c('magenta'), c('red')])

# Turn data points into rgb values
z_data_rgb = rgb(norm(phase))
# normalizing the intensity values
intensity = norm(mag)

# defining light source
ls = mcolors.LightSource()

# plotting
plt.imshow(ls.blend_hsv(z_data_rgb, intensity), extent=[-3, 3, -3, 3])
plt.show()

我得到以下图表: 如果它工作正确，基于强度值，一些区域的饱和度应该比其他区域低。

谢谢

- Neill Herbst

Answer 1

为什么不使用 hsv_to_rgb 并用 rgb 颜色绘图呢？

from matplotlib.colors import hsv_to_rgb
rgb = hsv_to_rgb(hsv)

- Serenity