我想用 seaborn 進行可視化并添加文本。這是我的代碼：# barplot price by body-stylefig, ax = plt.subplots(figsize = (12,8))g = data[['body-style','price']].groupby(by = 'body- style').sum().reset_index().sort_values(by='price')x = g['body-style']y = g['price']ok = sns.barplot(x,y, ci = None)ax.set_title('Price By Body Style')def autolabel(rects): for idx,rect in enumerate(ok): height = rect.get_height() g.text(rect.get_x() + rect.get_width()/2., 0.2*height, g['price'].unique().tolist()[idx], ha='center', va='bottom', rotation=90)autolabel(ok)但我出錯了：
1 回答
一只甜甜圈
TA貢獻1836條經(jīng)驗 獲得超5個贊
你需要做一些改變：
由于您已經(jīng)創(chuàng)建了
ax，因此您需要sns.barplot(..., ax=ax).autolabel()需要以柱列表作為參數(shù)調(diào)用。使用 seaborn，您可以通過ax.patches.for idx,rect in enumerate(ok):不應該使用ok但是rects。你不能使用
g.text。g是一個數(shù)據(jù)框，沒有功能.text。你需要ax.text。用作
g['price'].unique().tolist()[idx]要打印的文本與繪制的條形圖沒有任何關系。你可以改用height。
這是一些帶有玩具數(shù)據(jù)的測試代碼：
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
fig, ax = plt.subplots(figsize=(12, 8))
g = data[['body-style','price']].groupby(by = 'body-style').sum().reset_index().sort_values(by='price')
x = g['body-style']
y = g['price']
# x = list('abcdefghij')
# y = np.random.randint(20, 100, len(x))
sns.barplot(x, y, ci=None, ax=ax)
ax.set_title('Price By Body Style')
def autolabel(rects):
for rect in rects:
height = rect.get_height()
ax.text(rect.get_x() + rect.get_width() / 2., 0.2 * height,
height,
ha='center', va='bottom', rotation=90, color='white')
autolabel(ax.patches)
plt.show()
PS：您可以通過參數(shù)將文本的字體大小更改為ax.text: ax.text(..., fontsize=14)。
添加回答
舉報
0/150
提交
取消