Python 在绘图中应用条件不透明度

解

我首先为类别创建了一个RGB colored颜色 映射，从而获得了我想要的解决方案.然后，我通过创建相应的RGBA值并添加基于selected列的相关不透明度，将不透明度添加到 colored颜色 贴图中.

import colorsys
import pandas as pd
import plotly.graph_objects as go

def create_color_list(df: pd.DataFrame) -> List[str]:
    
    unique_categories = df['category'].unique()
    category_color_map = {}

    num_categories = len(unique_categories)
    color_map = []
    for i in range(num_categories):
        hue = i / num_categories
        rgb = tuple(round(i * 255) for i in colorsys.hsv_to_rgb(hue, 0.7, 0.9))
        color_map.append(rgb)

    for i, category in enumerate(unique_categories):
        category_color_map[category] = color_map[i % len(color_map)]

    color_list = []
    for category, selected in df[['category', 'selected']].values:
        if selected:
            opacity = 0.8
        else:
            opacity = 0.2
        color = category_color_map[category]
        rgba = f"rgba({color[0]}, {color[1]}, {color[2]}, {opacity})"
        color_list.append(rgba)
        
    return color_list


def build_product_data_fig(df: pd.DataFrame, color_list: List[str]) -> go.Figure:

    product_fig = go.Figure()
    product_fig.add_trace(go.Scatter(mode='markers',
                                     x=df['emission'],
                                     y=df['weight_gram'],
                                     marker=dict(
                                         color=color_list,
                                         size=df['price'] / 40)
                                     ))

    product_fig.update_layout(template='plotly_dark')

    return product_fig

color_list = create_color_list(df)
product_fig = build_product_data_fig(df, color_list)
product_fig.show()