overlay with filtered contours
"y_rotated"
]
},
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "num_contours = 20\n",
+ "\n",
+ "diff = np.ceil(np.nanmax(treatment_dose) / ((num_contours - 1) * 10)) * 10\n",
+ "levels = np.arange(0, diff * num_contours, diff)\n",
+ "levels"
+ ]
+ },
{
"cell_type": "code",
"execution_count": null,
"outputs": [],
"source": [
"plt.figure(figsize=(20,20))\n",
- "plt.contourf(x_rotated, y_rotated, treatment_dose, 20)\n",
+ "plt.contourf(x_rotated, y_rotated, treatment_dose, levels)\n",
"plt.colorbar()\n",
"\n",
+ "plt.contour(x_rotated, y_rotated, filtered_dose, levels=levels, colors='k')\n",
+ "\n",
+ "\n",
"plt.plot([0, 0], [-90, 50])\n",
"plt.plot([-60, 60], [0, 0])\n",
"plt.plot(0, 0, 'o', markeredgecolor='k', markersize=20)\n",
