I am trying to create a scatter plot for a csv file in python which contains 4 columns x
, y
, TL
, L
. I am supposed to plot x
versus y
and change the color of the marker based on the class ID
in the TL
column which I have achieved in the below code.
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()
Also, I need to change the shape depending on whether the marker has a label in the L
column or not, but I dont know how to update my code to fit this requirement.
Here is the link for the knnDataSet.csv
file:
knnDataSet.csv
See Question&Answers more detail:
os 与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…