下面是我想要绘制的散点图的代码。我想在绘制子图后添加相关线。虽然代码运行没有错误,但我看不到绘图中绘制的任何线条。任何帮助都感激不尽。
fig, ax = plt.subplots(figsize=fs)
for g in np.unique(group):
ix = np.where(group == g)
sc = ax.scatter(y[ix], dosage[ix], color = cdict[g], edgecolors='none')
# Fit with polyfit
b, m = polyfit(y, dosage, 1)
ax.plot(y, b + m * y, '-')
ax.set_xlabel('LT2', fontsize = 16)
ax.set_ylabel('Log Dosage', fontsize = 16)
# ax.legend(scatterpoints = 1, title='DILI Risk')
# fig.colorbar(sc, label='Dosage/mg/day')
path_dose = 'figure/reports/images/{}_hist_dose.png'.format(name_rep)
data['hist_dosage'].append(path_dose)
plt.savefig(path_dose)
plt.close()
对不起,我的数据中有nans。 np.polyfit()在第一时间没有工作,因此没有任何可绘制的内容。