我有一组应用程序上行链路/下行链路,分布显示为CDF,我想通过在其图例中添加更多信息来改进绘图,例如,我希望它向我显示数字每个变量的点, 对于这段代码,而不是向我展示:
--- Uplink
___ Downlink
我想看到这个:
--- Uplink(54)
___ Downlink(88)
其中54是此上行链路中绘制点的数量,其中88是此下行链路中绘制点的数量
以下是我想要改进的代码:
import csv
import matplotlib.pyplot as plt
import pylab
#---Reading from a csv file
def getColumn(filename, column):
results = csv.reader(open(filename), delimiter="\t")
return [result[column] for result in results]
#---the cdf Function
def cdf(inputList):
outputList = []
inputList.sort()
for i in range(0, len(inputList)):
outputList.append( [inputList[i], (i+1.0)/len(inputList)] )
return outputList
#--getting the data i want from the csv file
fgfS=getColumn("dataFG.csv",2)
bgfS=getColumn("dataBG.csv",2)
fgifS=getColumn("dataiFG.csv",2)
bgifS=getColumn("dataiBG.csv",2)
#---the points to plot:
fgfS1=[]
fgfS2=[]
for i in cdf(map(float,fgfS)):
fgfS1.append(i[0])
fgfS2.append(i[1])
bgfS1=[]
bgfS2=[]
for i in cdf(map(float,bgfS)):
bgfS1.append(i[0])
bgfS2.append(i[1])
fgifS1=[]
fgifS2=[]
for i in cdf(map(float,fgifS)):
fgifS1.append(i[0])
fgifS2.append(i[1])
bgifS1=[]
bgifS2=[]
for i in cdf(map(float,bgifS)):
bgifS1.append(i[0])
bgifS2.append(i[1])
fig2, (ax2,ax02) = plt.subplots(1,2, sharey=True)
labels=('Uplink','Downlink')
fig2.suptitle('Applications Down/Up link', fontsize=20)
#******************
ax2.set_title('App 1')
#ax2.set_xlabel('seconds')
ax2.set_ylabel('CDF')
ax2.grid(True)
ax2.plot(fgfS1,fgfS2, 'b')
ax2.plot(bgfS1,bgfS2, 'c')
ax2.set_xscale('log')
ax2.legend(labels,loc='best')
#******************
ax02.set_title('App 3')
ax02.set_xlabel('seconds')
ax02.set_ylabel('CDF')
ax02.grid(True)
ax02.plot(fgifS1,fgifS2, 'b')
ax02.plot(bgifS1,bgifS2, 'c')
ax02.set_xscale('log')
ax02.legend(labels,loc='best')
所以我想添加len(bgfS1),但我仍然没有弄清楚在哪里以及如何....
答案 0 :(得分:0)
只需将其附加到labels
列表组件,即可确定最终显示的内容。
所以你想要:
ax2.legend(['{} ({})'.format(lab, cnt) for lab,cnt in izip(labels, (len(fgfS1), len(bgfS1)))],loc='best')
(这实际上只是将计数附加到每个标签上)