Question

我有一组应用程序上行链路/下行链路，分布显示为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），但我仍然没有弄清楚在哪里以及如何....

Answer 1

只需将其附加到labels列表组件，即可确定最终显示的内容。

所以你想要：

ax2.legend(['{} ({})'.format(lab, cnt) for lab,cnt in izip(labels, (len(fgfS1), len(bgfS1)))],loc='best')

（这实际上只是将计数附加到每个标签上）

使用Matplotlib向图例添加信息

1 个答案: