为 plotly express 带状图中的特定点着色

Color specific points in a plotly express strip plot

让我们使用这个示例数据框:

import random

df=pd.DataFrame({'Id':list(range(1,101)), 'Value':[random.randint(0,50) for i in range(100)]})

我想创建一个 plotly express strip plot,其中 id 5 和 10 为红色,其他为蓝色。我构建了以下函数:

import plotly.express as px

def strip_plotly(df,col_value,col_hover=[],col_color=None,L_color=[]):
        
    if col_color is not None :
        L_color = ["red" if i in L_color else "blue" for i in df[col_color].values]
    else :
        L_color = ["blue"]*len(df.index)
        
    fig = px.strip(df, y=col_value,hover_data = col_hover,color_discrete_sequence = L_color)

    fig.update_layout({
            'plot_bgcolor': 'rgba(0,0,0,0)',
            'paper_bgcolor': 'rgba(0,0,0,0)',
        },
        hoverlabel=dict(
            #bgcolor="white", 
            font_size=12, 
            #font_family="Rockwell"
        ),
        xaxis={
            'title':"x",
            #'type':'log'
        },
        yaxis={'title':"y"},
        title={
            'text': "strip plot for Whosebug",
            #'y':0.95,
            'x':0.5,
            'xanchor': 'center',
            'yanchor': 'top'}
    )

    fig.update_xaxes(showspikes=True, spikecolor = 'black', showline=True, linewidth=1,linecolor='black',
                     ticks = "outside", tickwidth = 1, tickcolor = 'black',ticklen = 5)
    fig.update_yaxes(showspikes=True, spikecolor = 'black', showline=True, linewidth=1,linecolor='black',
                     ticks = "outside", tickwidth = 1, tickcolor = 'black',ticklen = 5)
    
    fig.show()

但是当我运行函数时:

strip_plotly(df,"Value",col_hover=["Id"],col_color="Id",L_color=[5,10])

我只得到蓝色点(id 5 和 10 不是红色)。你能解释一下哪里出了问题吗?

import random
import plotly.express as px
import numpy as np
import pandas as pd

df=pd.DataFrame({'Id':list(range(1,21)), 'Value':[random.randint(0,50) for i in range(20)]})


def strip_plotly(df,col_value,col_hover=[],col_color=None,L_color=[]):
    L_color = np.where(df["Id"].isin([5,10]), "red", "blue")

    # need two traces as color is in marker https://plotly.github.io/plotly.py-docs/generated/plotly.graph_objects.box.marker.html
    fig = px.strip(df.loc[L_color=="blue"], y=col_value,hover_data = col_hover,color_discrete_sequence = ["blue"])
    fig = fig.add_traces(px.strip(df.loc[L_color=="red"], y=col_value,hover_data = col_hover,color_discrete_sequence = ["red"]).data)
    fig = fig.update_traces(offsetgroup="1")

    fig.update_layout({
            'plot_bgcolor': 'rgba(0,0,0,0)',
            'paper_bgcolor': 'rgba(0,0,0,0)',
        },
        hoverlabel=dict(
            #bgcolor="white", 
            font_size=12, 
            #font_family="Rockwell"
        ),
        xaxis={
            'title':"x",
            #'type':'log'
        },
        yaxis={'title':"y"},
        title={
            'text': "strip plot for Whosebug",
            #'y':0.95,
            'x':0.5,
            'xanchor': 'center',
            'yanchor': 'top'}
    )

    fig.update_xaxes(showspikes=True, spikecolor = 'black', showline=True, linewidth=1,linecolor='black',
                     ticks = "outside", tickwidth = 1, tickcolor = 'black',ticklen = 5)
    fig.update_yaxes(showspikes=True, spikecolor = 'black', showline=True, linewidth=1,linecolor='black',
                     ticks = "outside", tickwidth = 1, tickcolor = 'black',ticklen = 5)
    
    return fig.show()
    
strip_plotly(df,"Value",col_hover=["Id"],col_color="Id",L_color=[5,10])