|
63 | 63 | "source": [
|
64 | 64 | "import numpy as np\n",
|
65 | 65 | "from matplotlib import pyplot as plt\n",
|
66 |
| - "from scipy.stats import loglaplace,chi2\n", |
| 66 | + "from scipy.stats import loglaplace, chi2\n", |
67 | 67 | "\n",
|
68 | 68 | "from labellines import labelLine, labelLines"
|
69 | 69 | ]
|
|
125 | 125 | ],
|
126 | 126 | "source": [
|
127 | 127 | "for a in A:\n",
|
128 |
| - " plt.plot(X, np.arctan(a*X), label=str(a))\n", |
| 128 | + " plt.plot(X, np.arctan(a * X), label=str(a))\n", |
129 | 129 | "\n",
|
130 | 130 | "labelLines(plt.gca().get_lines(), zorder=2.5)"
|
131 | 131 | ]
|
|
171 | 171 | ],
|
172 | 172 | "source": [
|
173 | 173 | "for a in A:\n",
|
174 |
| - " plt.plot(X, np.sin(a*X), label=str(a))\n", |
| 174 | + " plt.plot(X, np.sin(a * X), label=str(a))\n", |
175 | 175 | "\n",
|
176 | 176 | "labelLines(plt.gca().get_lines(), align=False, fontsize=14)"
|
177 | 177 | ]
|
|
217 | 217 | ],
|
218 | 218 | "source": [
|
219 | 219 | "for a in A:\n",
|
220 |
| - " plt.plot(X, loglaplace(4).pdf(a*X), label=str(a))\n", |
| 220 | + " plt.plot(X, loglaplace(4).pdf(a * X), label=str(a))\n", |
221 | 221 | "\n",
|
222 | 222 | "xvals = [0.8, 0.55, 0.22, 0.104, 0.045]\n",
|
223 |
| - "labelLines(plt.gca().get_lines(), align=False, xvals=xvals, color='k')" |
| 223 | + "labelLines(plt.gca().get_lines(), align=False, xvals=xvals, color=\"k\")" |
224 | 224 | ]
|
225 | 225 | },
|
226 | 226 | {
|
|
263 | 263 | ],
|
264 | 264 | "source": [
|
265 | 265 | "for a in A:\n",
|
266 |
| - " plt.plot(X, chi2(5).pdf(a*X), label=str(a))\n", |
| 266 | + " plt.plot(X, chi2(5).pdf(a * X), label=str(a))\n", |
267 | 267 | "\n",
|
268 | 268 | "lines = plt.gca().get_lines()\n",
|
269 |
| - "l1=lines[-1]\n", |
270 |
| - "labelLine(l1, 0.6, label=r'$Re=${}'.format(l1.get_label()), ha='left',\n", |
271 |
| - " va='bottom', align=False)\n", |
| 269 | + "l1 = lines[-1]\n", |
| 270 | + "labelLine(l1, 0.6, label=rf\"$Re=${l1.get_label()}\", ha=\"left\", va=\"bottom\", align=False)\n", |
272 | 271 | "labelLines(lines[:-1], align=False)"
|
273 | 272 | ]
|
274 | 273 | },
|
|
313 | 312 | ],
|
314 | 313 | "source": [
|
315 | 314 | "for a in A:\n",
|
316 |
| - " plt.semilogx(X, np.arctan(5*a*X), label=str(a))\n", |
| 315 | + " plt.semilogx(X, np.arctan(5 * a * X), label=str(a))\n", |
317 | 316 | "\n",
|
318 | 317 | "labelLines(plt.gca().get_lines(), zorder=2.5)"
|
319 | 318 | ]
|
|
359 | 358 | ],
|
360 | 359 | "source": [
|
361 | 360 | "for a in A:\n",
|
362 |
| - " plt.semilogx(X, chi2(5).pdf(a*X), label=str(a))\n", |
| 361 | + " plt.semilogx(X, chi2(5).pdf(a * X), label=str(a))\n", |
363 | 362 | "\n",
|
364 | 363 | "labelLines(plt.gca().get_lines(), xvals=(0.1, 1), zorder=2.5)"
|
365 | 364 | ]
|
|
396 | 395 | "for j, ax in enumerate(axes):\n",
|
397 | 396 | " fig.sca(ax)\n",
|
398 | 397 | " for i in range(1, 5):\n",
|
399 |
| - " plt.plot(X, X**(-2)/i, label='$x^2/%s$' % i)\n", |
| 398 | + " plt.plot(X, X ** (-2) / i, label=f\"$x^2/{i}$\")\n", |
400 | 399 | " plt.ylim(0, 10)\n",
|
401 | 400 | "\n",
|
402 | 401 | " lines = []\n",
|
403 | 402 | " for i in range(-8, 12, 2):\n",
|
404 |
| - " lines.extend(plt.plot(X, 10*X + i, '--', color='gray', label='i=%s' % i))\n", |
| 403 | + " lines.extend(plt.plot(X, 10 * X + i, \"--\", color=\"gray\", label=f\"i={i}\"))\n", |
405 | 404 | "\n",
|
406 | 405 | " if j == 0:\n",
|
407 | 406 | " labelLines(lines, zorder=2.5)\n",
|
408 |
| - " plt.title('$\\mathtt{drop\\_label=False}$')\n", |
| 407 | + " plt.title(\"$\\mathtt{drop\\_label=False}$\")\n", |
409 | 408 | " else:\n",
|
410 | 409 | " labelLines(lines, zorder=2.5, drop_label=True)\n",
|
411 |
| - " plt.title('$\\mathtt{drop\\_label=True}$')\n", |
| 410 | + " plt.title(\"$\\mathtt{drop\\_label=True}$\")\n", |
412 | 411 | " plt.legend()"
|
413 | 412 | ]
|
414 | 413 | },
|
|
0 commit comments