在3D散点图中忽略了我的自定义颜色

Question

我正在渲染一个3D空间，看起来像一半的球体，使用图中的3D散点图。我想根据熊猫数据框架中的值更改每个点的颜色。

我的数据框的行看起来像这样：

{
   "x": 1,
   "y": -31,
   "z": -63,
   "M": -1,
   "E": -1,
   "F": -1,
   "CE": -1,
   "MA": -1,
   "R": -1
}

这样，为了构建我使用的图表：

fig = px.scatter_3d(json_df, x="x", y="y", z="z", custom_data=["M", "E", "F", "CE", "MA", "R"])
fig.update_traces(
    hovertemplate="<br>".join([
        "M: %{customdata[0]}",
        "E: %{customdata[1]}",
        "F: %{customdata[2]}",
        "CE: %{customdata[3]}",
        "MA: %{customdata[4]}",
        "R: %{customdata[5]}"
    ])
)
fig.show()

它会产生此图像（如果您也悬停了每个点，则在HoverTemplate中获得信息，

现在我想根据这些值m，e，f等更改每个点的颜色，对于非valid点具有“默认颜色”（其中m，e，f的那些等。是-1）。目前，我想对非浮游点使用颜色，而有效的颜色则是随机色。

灰色

grey_colour = 'rgb(84, 84, 84)'
def row_to_color(df_row):
    if df_row["M"] == -1:
        # non-valid colour
        return grey_colour 
    else:
        #randomly selected colour
        return 'rgb(' + str(0) + ', ' + str(np.random.randint(0,255)) + ', ' + str(np.random.randint(0,255)) + ')'

# Almost 2000-elements-long array, with a large part of it being that grey default color. 
custom_color = json_df.apply(row_to_color, axis=1)
fig = px.scatter_3d(json_df, x="x", y="y", z="z", custom_data=["M", "E", "F", "CE", "MA", "R"], color=custom_color)

但是，绘制忽略了该非valid颜色（无论是什么），然后返回到默认的颜色，只尊重与默认的颜色不同的颜色。

我已经进行了一些测试，如果我使用其他条件（因此，Custom_color数组中的“非valid颜色”项目少得多），则该图似乎正确地呈现，但是颜色为仍然不正确。例如：

def row_to_color(df_row):
    if df_row["x"] > 10: #less strict condition
        return 'rgb(84, 84, 84)' #grey
    else:
        return 'rgb(' + str(0) + ', ' + str(np.random.randint(0,255)) + ', ' + str(np.random.randint(0,255)) + ')'
...

fig = px.scatter_3d(...)

浅橙色？？？？

对正在发生的事情有任何想法吗？

Answer 1

和 Plotly的正式文档，不幸的是，我已经误读了），如果您想为最终图像中要绘制的每个项目使用特定颜色，则只需使用color_discrete_map =“ Identity”时，您的情节功能。

如果您不使用它，则Plotly试图使用color数组中指定的颜色作为要分配给类别的颜色的颜色。如果您没有类别，则在绘图过程中，情节只是发疯了。

通过将该参数添加到呼叫中，我们会告诉图片分别处理color数组中的每个项目，并将其设置为所提供数据中相应项目的颜色。

fig = px.scatter_3d(json_df, x="x", y="y", z="z", 
    custom_data=["M", "E", "F", "CE", "MA", "R"], 
    color=custom_color,
    color_discrete_map="identity")