如何识别图形线条

我有以下格式的路径的 x、y 数据（示例仅用于说明）：

   seq    p1      p2
0  20      2      3
1  20      2      4
2  20      4      4
3  22      5      5
4  22      5      6
5  23      6      2
6  23      6      3
7  23      6      4

每条路径都有多个点，它们由 seq 标识，属于同一 seq 的点被认为是一条路径，依此类推。

我已经使用以下代码绘制了这些路径（使用与上面格式相同的真实数据），并附上了结果：

import matplotlib.pyplot as plt

fig, ax = plt.subplots(figsize=(12, 8))

for (key, grp) in df.groupby("seq"):
    grp.plot(linestyle = "solid", x="p1", y="p2", ax = ax, label = key)

box = ax.get_position()
ax.set_position([box.x0, box.y0, box.width * 0.8, box.height])
ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))

plt.title("Paths")
plt.show()

我已经绘制了大约 40 条路径，现在的问题是我应该如何识别哪条路径是针对 uid 184 的，或者哪条路径是针对 uid-194 的？它们在图例中都用相同的颜色标记。有没有一种方法可以让我能够独特地识别每条路径，也许可以在路径上的某个位置进行标记（但这可能会使图形变得混乱）。

我的第二个问题是我想标记每条路径/轨迹的起点和终点。喜欢起点 can be green和终点 can be red。例如，在上面的示例 df 中，对于 uid-20，起点是第 0 行中的 (2,3)，终点是第 2 行中的 (4,4)。请建议一种标记这些起点和终点的方法df 中的每条路径。

我不确定这在多大程度上回答了您的问题，但这是一种使线条颜色符合颜色图的方法，这通常可以帮助我更轻松地可视化不同的线条及其趋势，但可能不太方便挑选出单条线条。

cmap = plt.cm.get_cmap('viridis')
groups = df.groupby("uid")
ngroups = len(groups)

for i, (key, grp) in enumerate(groups):
    grp.plot(linestyle="solid", x="px", y="py", ax=ax, label=key, color=cmap(i/(ngroups-1))

要为第一个点和最后一个点添加标记，只需将它们挑出来并为它们分配您喜欢的颜色和标记。因此，重写上面的 for 循环：

for i, (key, grp) in enumerate(df.groupby("uid")):
    grp.plot(linestyle="solid", x="px", y="py", ax=ax, label=key, color=cmap(i/(ngroups-1)))
    grp.iloc[[0]].plot(marker="o", x="px", y="py", ax=ax, color='r', legend=False)
    grp.iloc[[-1]].plot(marker="o", x="px", y="py", ax=ax, color='g', legend=False)

如果您希望每个标记具有不同的红色和绿色色调，您可以使用Reds and Greens颜色图例如：

start_cmap =  plt.cm.get_cmap('Reds')
end_cmap =  plt.cm.get_cmap('Greens')

在循环中它将是

grp.iloc[[0]].plot(marker="o", x="px", y="py", ax=ax, color=start_cmap(i/(ngroups-1)), legend=False)
grp.iloc[[-1]].plot(marker="o", x="px", y="py", ax=ax, color=end_cmap(i/(ngroups-1)), legend=False)

EDITS

处理图例

为了仅绘制线条而不绘制端点标记，我们首先绘制了线条，然后绘制了两个标记，这就是将绘图推入轴线队列的方式，因此我们跳过标记并明确告诉图例要考虑哪些行：

ax.legend(ax.lines[::3], groups.groups.keys(), loc='center left', bbox_to_anchor=(1, 0.5))

使用颜色条

如果对线条使用颜色图，则显示颜色条而不是图例很有用，因此我们使用如下所示的内容：

from matplotlib.colorbar import ColorbarBase
import matplotlib as mp

values = list(groups.groups.keys())
cax = fig.add_axes([0.92, 0.12, 0.02, 0.75])
cbar = ColorbarBase(cax, cmap=cmap, format='%d', ticks=values, drawedges=False, norm=mp.colors.Normalize(vmin=min(values), vmax=max(values)))