Question

我想修改下图所示的Y轴单位。对于大数字，优选使用 M （百万）， k （千）等单位。例如，y轴应该看起来像：50k，100k，150k等。

以下图表由以下代码段生成：

int minIndex = 0;
for (int k = 1; k < i; k++) {
    if (scores[minIndex] > scores[k]) {
        minIndex = k;
    }
}
double lowestScore = scores[minIndex];

我看到this帖子，并认为我可以编写自己的格式化功能：

plt.autoscale(enable=True, axis='both')
plt.title("TTL Distribution")
plt.xlabel('TTL Value')
plt.ylabel('Number of Packets')
y = graphy  # data from a sqlite query
x = graphx  # data from a sqlite query
width = 0.5
plt.bar(x, y, width, align='center', linewidth=2, color='red', edgecolor='red')
fig = plt.gcf()
plt.show()

但我错过了我正在使用的条形图没有def y_fmt(x, y): if max_y > 1000000: val = int(y)/1000000 return '{:d} M'.format(val) elif max_y > 1000: val = int(y) / 1000 return '{:d} k'.format(val) else: return y功能。

如何更好地格式化Y轴？

[ TTL Distribution Plot ]

Answer 1

原则上，始终可以选择通过plt.gca().yaxis.set_xticklabels()设置自定义标签。

但是，我不确定为什么不应该在这里使用matplotlib.ticker.FuncFormatter。 FuncFormatter的设计目的是根据ticklabel的位置和值提供自定义ticklabel。 matplotlib示例集合中实际上有nice example。

在这种情况下，我们可以根据需要使用FuncFormatter在matplotlib图的轴上提供单位前缀作为后缀。为此，我们迭代1000的倍数并检查要格式化的值是否超过它。如果该值是整数，我们可以将其格式化为整数，并将相应的单位符号作为后缀。另一方面，如果小数点后面有余数，我们会检查格式化这个数字需要多少小数位。

这是一个完整的例子：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.ticker import FuncFormatter

def y_fmt(y, pos):
    decades = [1e9, 1e6, 1e3, 1e0, 1e-3, 1e-6, 1e-9 ]
    suffix  = ["G", "M", "k", "" , "m" , "u", "n"  ]
    if y == 0:
        return str(0)
    for i, d in enumerate(decades):
        if np.abs(y) >=d:
            val = y/float(d)
            signf = len(str(val).split(".")[1])
            if signf == 0:
                return '{val:d} {suffix}'.format(val=int(val), suffix=suffix[i])
            else:
                if signf == 1:
                    print val, signf
                    if str(val).split(".")[1] == "0":
                       return '{val:d} {suffix}'.format(val=int(round(val)), suffix=suffix[i]) 
                tx = "{"+"val:.{signf}f".format(signf = signf) +"} {suffix}"
                return tx.format(val=val, suffix=suffix[i])

                #return y
    return y


fig, ax = plt.subplots(ncols=3, figsize=(10,5))

x = np.linspace(0,349,num=350) 
y = np.sinc((x-66.)/10.3)**2*1.5e6+np.sinc((x-164.)/8.7)**2*660000.+np.random.rand(len(x))*76000.  
width = 1

ax[0].bar(x, y, width, align='center', linewidth=2, color='red', edgecolor='red')
ax[0].yaxis.set_major_formatter(FuncFormatter(y_fmt))

ax[1].bar(x[::-1], y*(-0.8e-9), width, align='center', linewidth=2, color='orange', edgecolor='orange')
ax[1].yaxis.set_major_formatter(FuncFormatter(y_fmt))

ax[2].fill_between(x, np.sin(x/100.)*1.7+100010, np.cos(x/100.)*1.7+100010, linewidth=2, color='#a80975', edgecolor='#a80975')
ax[2].yaxis.set_major_formatter(FuncFormatter(y_fmt))

for axes in ax:
    axes.set_title("TTL Distribution")
    axes.set_xlabel('TTL Value')
    axes.set_ylabel('Number of Packets')
    axes.set_xlim([x[0], x[-1]+1])

plt.show()

提供以下图表：

Answer 2

你非常接近;关于FuncFormatter的一个（可能）令人困惑的事情是第一个参数是刻度值，第二个是刻度位置，（当命名为x,y时）可能会混淆y轴。为清楚起见，我在下面的示例中重命名了它们。

该函数应该接受两个输入（刻度值x和位置pos）并返回一个字符串

（http://matplotlib.org/api/ticker_api.html#matplotlib.ticker.FuncFormatter）

工作示例：

import numpy as np
import matplotlib.pylab as pl
import matplotlib.ticker as tick

def y_fmt(tick_val, pos):
    if tick_val > 1000000:
        val = int(tick_val)/1000000
        return '{:d} M'.format(val)
    elif tick_val > 1000:
        val = int(tick_val) / 1000
        return '{:d} k'.format(val)
    else:
        return tick_val

x = np.arange(300)
y = np.random.randint(0,2000000,x.size)

width = 0.5
pl.bar(x, y, width, align='center', linewidth=2, color='red', edgecolor='red')
pl.xlim(0,300)

ax = pl.gca()
ax.yaxis.set_major_formatter(tick.FuncFormatter(y_fmt))

Matplotlib / pyplot：自动调整y轴的单位

2 个答案: