使用Matplotlib向图例添加信息

2024-04-29 01:19:49 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个图,一组应用程序的上行链路/下行链路,分布用CDF表示,我想通过在其图例中添加更多信息来改进绘图,例如,我希望它显示每个变量的点数, 对于这个代码,而不是给我看:

--- Uplink
___ Downlink

我想看看这个:

^{pr2}$

其中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)作为例子,但我还是不知道在哪里以及如何。。。。在


Tags: csvtheinmapforplotfloat链路
1条回答
网友
1楼 · 发布于 2024-04-29 01:19:49

只需将它附加到您的labels列表组件,这些组件决定最终将显示什么。在

所以你需要:

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

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

相关问题 更多 >