借助DataFrame的pivot函数进行数据的可视化
这是数据

1.把数据读取成DataFrame格式
import pandas as pd
import matplotlib.pyplot as plt
df = pd.read_csv('score(1).csv',header=None)
data_original = {'algorithm':df.iloc[1:,1].values,'target':df.iloc[1:,2].values,'number':df.iloc[1:,3].values}
data_original = pd.DataFrame(data_original)
data_original
2.使用pivot
函数对DataFrame进行数据透视
data = data_original.pivot(index = 'algorithm',columns = 'target',values = 'number')
data
3.我们只需要数据,所以
data.values
4. 画图
plt.plot(range(5), data.values[0,:], 'r-',label = 'relief_f')
plt.plot(range(5), data.values[1,:], 'g-',label = 'relief_f_n')
plt.xticks(range(5),data.columns)
plt.xlabel('Target')
plt.ylabel('Precision')
plt.title("The Precision")
ax = plt.gca()
ax.invert_yaxis() #y轴反向
plt.legend() # 显示左下角的图例
plt.show()
每篇小附录:
SNAP图论数据集