Я хотел бы использовать два цвета, красный и синий, но с разной концентрацией, как показано ниже. Я хотел бы преобразовать эту непрерывную цветовую шкалу в дискретную цветовую шкалу с 10 дискретными цветами.
https://plotly.com/python/colorscales/#reversing-a-builtin-color-scale
Если я печатаю непрерывную цветовую шкалу, в списке есть только 2 элемента, как показано ниже. Теперь, как я могу получить 10 дискретных цветов между красным и синим с разной концентрацией. Спасибо
colors=px.colors.sequential.Bluered_r
print(colors)
['rgb(255,0,0)', 'rgb(0,0,255)']
Насколько я знаю, сюжетно не имеет для этого явной функции. Для красно-синей шкалы должна работать простая реализация на основе np.linspace.
import numpy as np
def n_discrete_rgb_colors(color1: str, color2: str, n_colors: int) -> list:
color1_ = [int(i) for i in color1[4:-1].split(",")]
color2_ = [int(i) for i in color2[4:-1].split(",")]
colors_ = np.linspace(start = color1_, stop = color2_, num = n_colors)
colors = [str(f"rgb{int(i[0]),int(i[1]),int(i[2])}") for i in colors_]
return colors
color1, color2 = ['rgb(255,0,0)', 'rgb(0,0,255)']
# color1, color2 = px.colors.sequential.Bluered_r
n_discrete_rgb_colors(color1, color2, 10)
Выход
['rgb(255, 0, 0)',
'rgb(226, 0, 28)',
'rgb(198, 0, 56)',
'rgb(170, 0, 85)',
'rgb(141, 0, 113)',
'rgb(113, 0, 141)',
'rgb(85, 0, 170)',
'rgb(56, 0, 198)',
'rgb(28, 0, 226)',
'rgb(0, 0, 255)']
Логика кода следующая:
'rgb(x,y,z)'
,[x,y,z]
списки,import pandas as pd
import numpy as np
import plotly.express as px
df = pd.DataFrame({c:np.linspace(1,10,100) for c in list("xyc")})
# https://plotly.com/python/colorscales/#constructing-a-discrete-or-discontinuous-color-scale
cmap = [
(r, c)
for r, c in zip(
np.repeat(np.linspace(0, 1, len(px.colors.sequential.RdBu)+1), 2)[1:],
np.repeat(px.colors.sequential.RdBu,2),
)
]
px.bar(df, x = "x", y = "y", color = "c", color_continuous_scale=cmap)
Спасибо Константин. Это тоже работает. Я надеюсь, что смогу принять два ответа. я ценю его