[python] density plot에서 Peak 값(max or min value) 구하고 표시하기
목차
fig1. title
#1. kdeplot
#2. get x, y
#3. get max or min
#4. marking
import seaborn as sns
import numpy as np
#1.
ax = sns.kdeplot(th_box_sub2, label='sub2')
#2.
x = ax.lines[0].get_xdata() # Get the x data of the distribution
y = ax.lines[0].get_ydata() # Get the y data of the distribution
#3.
maxid1 = np.argmax(y) # get max in all
maxid2 = np.argmax(y[x<0.59]) # get max in specific range
minid = np.argmin(y[(x>0.56) & (x<0.62)]) # get min in specific range
#4.
plt.plot(x[maxid1], y[maxid1], 'o', label=round(x[maxid1],2), ms=10)
plt.plot(x[maxid2], y[maxid2], 'v', label=round(x[maxid2],2), ms=10)
plt.plot(x[(x>0.56) & (x<0.62)][minid], y[(x>0.56) & (x<0.62)][minid], 'x', label=round(x[(x>0.56) & (x<0.62)][minid],2), ms=10)
plt.legend(fontsize=15)
plt.show()
