python - 如何根据列变量更改标记的形状？

Question

我正在尝试在 python 中为 csv 文件创建散点图，其中包含 4 列 x、y、TL、L。我应该绘制 x 与 y 的图，并根据 TL 中的 class ID 更改标记的颜色> 我在下面的代码中实现的列。

import pandas as pd    
from matplotlib import pyplot as plt    
import numpy as np

df=pd.read_csv('knnDataSet.csv')  
df.columns=['SN','x','y','TL','L']

color=['red','green','blue']

groups = df.groupby('TL')

fig, ax = plt.subplots()

for name, group in groups:    
    ax.plot(group.x, group.y, marker='o', linestyle='', ms=12, label=name)

ax.legend()    
plt.show()

此外，我需要根据标记在 L 列中是否有标签来更改形状，但我不知道如何更新我的代码以满足此要求。

以下是 knnDataSet.csv 文件的链接: knnDataSet.csv

Answer 1

你可能想要这样的东西:

import pandas as pd
from matplotlib import pyplot as plt
import numpy as np

df=pd.read_csv('knnDataSet.csv')
df.columns=['SN','x','y','TL','L']
color=['red','green','blue']

groups = df.groupby('TL')

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

for name, group in groups:   
    for x in group.values:
        if np.isnan(x[4]):
            ax.plot(x[1], x[2], marker='x', linestyle='', ms=12)
        else:
            ax.plot(x[1], x[2], marker='o', linestyle='', ms=12)                       

#ax.legend()
plt.show()

如果 L 列中的标签未定义，则标记将为 X
如果 L 列中的标签 IS 已定义，则标记将为 O

输出: