创建一个点/线的颜色依赖于另一列中的值的图？多种颜色类别

import matplotlib.pyplot as plt

# 假设有两列数据，x和color
x = [1, 2, 3, 4, 5]
color = [0, 1, 0, 1, 2]  # 假设color列的值为0、1、2，表示不同的颜色类别

# 创建散点图，颜色依赖于color列的值
plt.scatter(x, [0]*len(x), c=color, cmap='viridis')

# 设置颜色分类说明
plt.colorbar(ticks=[0, 1, 2], label='Color Category')

# 显示图形
plt.show()

import matplotlib.pyplot as plt

# 假设有两列数据，x和color
x = [1, 2, 3, 4, 5]
color = [0, 1, 0, 1, 2]  # 假设color列的值为0、1、2，表示不同的颜色类别

# 创建折线图，颜色依赖于color列的值
plt.plot(x, [0]*len(x), c=color, marker='o')

# 设置颜色分类说明
plt.colorbar(ticks=[0, 1, 2], label='Color Category')

# 显示图形
plt.show()