исключая нулевые значения для печати фигуры

Question

Я пытаюсь напечатать число с:

df_weight = df.groupby(['city'])['weight'].mean()

ax3 = df_weight.plot()
ax3.set_title("Weight by City");
ax3.figure.savefig("WeightCity.png")

Проблема в том, что у меня много нулевых значений в столбце веса, поэтому среднее значение неверно.

Я пытался это, чтобы найти значения «веса»> 0 и работал

weightUse = df.loc[df['weight'] >0 ,'weight']

, но я не знаю, как использовать эту информацию для печати фигуры. Есть идеи?

Answer 1

Для печати напрямую на стандартный вывод, вот способ:

import matplotlib.pyplot as plt
%matplotlib inline

df.query("weight > 0").groupby(['city'])['weight'].mean().plot()
plt.title("plot title")