检查pandas语句中的字符串并有条件地更改绘图颜色

2024-04-24 09:41:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在python中创建if循环的一些条件语句,以便有条件地在plotly中更改颜色。你知道吗

所以我想看看列alldf_na['x']中的1个是否包含字符串(A,B,C)。如果它们包含,颜色代码将是rgba(0,0,255, 0.8)。否则就是'rgba(255,255,0, 0.8)'. 代码如下:

node_trace = go.Scatter(
    x=[],
    y=[],
    text=[],
    mode='markers',
    hoverinfo='text',
    marker=dict(
        color=['rgba(0,0,255, 0.8)' if alldf_na['x'].str.contains(['A|B|C'], case=False, na=False).bool()  else 'rgba(255,255,0, 0.8)'],
        reversescale=True,
        size=[]),
        line=dict(width=2))

但是,存在以下错误:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Tags: 字符串代码textfalseif颜色语句plotly