本文共 1081 字，大约阅读时间需要 3 分钟。

Plotly Express：手动设置散点图点的颜色

在 Plotly Express 中，手动设置散点图中点的颜色可以通过以下步骤来实现

1. 导入必要的库

首先，需要导入 Plotly Express 库

import plotly.express as px

2. 准备数据集

接下来，准备一个包含所需颜色的数据集。以下是一个示例：

data = {    'x': [1, 2, 3, 4, 5],    'y': [6, 7, 8, 9, 10],    'color': ['red', 'blue', 'green', 'yellow', 'orange']}

在这个例子中，我们有一个包含 x、y 和 color 三个键的字典，其中 color 键对应的值是颜色列表。

3. 使用 Plotly Express 创建散点图

然后，使用 px.scatter() 函数创建散点图。通过将 color 参数设置为 color，可以实现手动设置颜色：

fig = px.scatter(data, x='x', y='y', color='color')

这样生成的散点图中，每个点的颜色将从 color 列表中随机分配。

4. 自定义每个点的颜色

如果你希望手动指定每个点的颜色，而不是从数据集中获取，可以直接在 color 参数中传入一个颜色列表。例如，如果你想让每个点都是红色：

fig = px.scatter(data, x='x', y='y', color=['red']*len(data['x']))

这样生成的散点图中，所有点的颜色都将设置为红色。

5. 测试用例

以下是一个完整的测试用例：

test_data = {    'x': [1, 2, 3, 4, 5],    'y': [6, 7, 8, 9, 10],    'color': ['red', 'blue', 'green', 'yellow', 'orange']}fig = px.scatter(test_data, x='x', y='y', color='color')fig.show()

运行上述代码后，你将看到一个包含不同颜色点的散点图。

6. 应用场景

在实际应用中，你可以将这些方法应用到更复杂的数据集中。例如，在机器学习中，你可以通过颜色来表示数据的类别或分布情况。

总结

通过以上方法，你可以轻松地在 Plotly Express 中手动设置散点图中点的颜色。无论是从数据集中获取颜色，还是手动指定每个点的颜色，这些方法都能满足你的需求。