如何在Seaborn点图上获取数据标签?

时间:2016-05-09 02:00:07

标签: python pandas matplotlib scipy seaborn

我有两个这样的数组:

Soldier_years = [1850, 1851, 1852, 1853, 1854, 1855, 1856, 1857, 1858, 1859, 1860, 1861, 1862, 1863, 1864, 1865, 1866, 1867, 1868, 1869, 1870]
num_records_yob = [7, 5, 8, 9, 15, 17, 23, 19, 52, 55, 73, 73, 107, 137, 65, 182, 228, 257, 477, 853, 2303]

我正试图把它们变成像Seaborn的那样:

%matplotlib inline
import seaborn as sns
import matplotlib.pyplot as plt
sns.set(style="darkgrid")

f, (ax) = plt.subplots(figsize=(12, 6), sharex=True)

sns.set_style("darkgrid")
ax = sns.pointplot(x=Soldier_years, y=num_records_yob)

我得到一个像这样的点图:

Pointplot

这个情节几乎就是我想要的。如何获得每个点的数据标签以显示在各个点之上?

我尝试了ax.patches,但它是空的。

我试图让它看起来像这样(但是对于点图): enter image description here

2 个答案:

答案 0 :(得分:8)

你可以这样做:

[ax.text(p[0], p[1]+50, p[1], color='g') for p in zip(ax.get_xticks(), num_records_yob)]

plot

答案 1 :(得分:0)

对于将来需要更一般性答案的参考,我建议使用以下代码:

ymin, ymax = ax.get_ylim()
color="#3498db" # choose a color
bonus = (ymax - ymin) / 50 # still hard coded bonus but scales with the data
for x, y, name in zip(X, Y, names):
    ax.text(x, y + bonus, name, color=color)

请注意,我也将理解改为for循环,我认为它更具可读性(当列表被实际抛弃时)