首页 > 解决方案 > 根据目标标签绘制每个特征的直方图

问题描述

我正在尝试为每个特征绘制直方图,并将类别分为目标值 0 和 1。我总共有 9 个功能。我想有类似的 输出

X=merged.iloc[:,0:10]
io=merged['target'][merged['target']==0]
nio=merged['target'][merged['target']==1]
fig,axes =plt.subplots(3,3, figsize=(12, 9))
ax=axes.ravel()

for i,column in enumerate(X.columns):
  _,bins=np.histogram(X[column],bins=40)
  ax[i].hist(io[column],bins=bins,color='r',alpha=.5)
  ax[i].hist(nio[column],bins=bins,color='g',alpha=0.3) 
  ax[i].axes.get_xaxis().set_visible(False)
  ax[i].set_yticks(())
ax[0].legend(['Negative','Positive'],loc='best',fontsize=8)
plt.tight_layout()
plt.show()

到目前为止,我已经尝试过了。但我遇到了一个关键错误。

标签: pythonmatplotlibdata-visualizationhistogram

解决方案


推荐阅读