fixed a wrong index

This commit is contained in:
philippun 2023-08-08 12:04:34 +02:00
parent 498f813d2d
commit ab08d5af1a

View File

@ -92,7 +92,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 16, "execution_count": 22,
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
@ -107,9 +107,9 @@
" r = row_index\n", " r = row_index\n",
" for i in range(2, cm.shape[0]-2):\n", " for i in range(2, cm.shape[0]-2):\n",
" j = offset\n", " j = offset\n",
" cm[i,j-1] = -s_x * alpha_interblock(alpha_x[r, j-1], alpha_x[r, j])\n", " cm[i,j-1] = -s_x * alpha_interblock(alpha_x[r, j-1-1], alpha_x[r, j-1])\n",
" cm[i,j] = 1 + s_x * (alpha_interblock(alpha_x[r,j], alpha_x[r, j+1]) + alpha_interblock(alpha_x[r, j], alpha_x[r, j-1]))\n", " cm[i,j] = 1 + s_x * (alpha_interblock(alpha_x[r,j-1], alpha_x[r, j+1-1]) + alpha_interblock(alpha_x[r, j-1], alpha_x[r, j-1-1]))\n",
" cm[i,j+1] = -s_x * alpha_interblock(alpha_x[r, j], alpha_x[r, j+1])\n", " cm[i,j+1] = -s_x * alpha_interblock(alpha_x[r, j-1], alpha_x[r, j+1-1])\n",
"\n", "\n",
" offset += 1\n", " offset += 1\n",
"\n", "\n",
@ -127,19 +127,27 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 17, "execution_count": 23,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
"ename": "IndexError", "name": "stdout",
"evalue": "index 5 is out of bounds for axis 1 with size 5", "output_type": "stream",
"output_type": "error", "text": [
"traceback": [ "[[ 0. 0. 0. 0. 0. 0.\n",
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m", " 0. ]\n",
"\u001b[0;31mIndexError\u001b[0m Traceback (most recent call last)", " [ 0. 0. 0. 0. 0. 0.\n",
"Cell \u001b[0;32mIn[17], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m result \u001b[39m=\u001b[39m create_coeff_matrix(np\u001b[39m.\u001b[39;49mrandom\u001b[39m.\u001b[39;49mrandom_sample((\u001b[39m5\u001b[39;49m,\u001b[39m5\u001b[39;49m)), \u001b[39m0\u001b[39;49m, \u001b[39m3\u001b[39;49m)\n\u001b[1;32m 2\u001b[0m \u001b[39mprint\u001b[39m(result)\n", " 0. ]\n",
"Cell \u001b[0;32mIn[16], line 13\u001b[0m, in \u001b[0;36mcreate_coeff_matrix\u001b[0;34m(alpha_x, row_index, s_x, transpose)\u001b[0m\n\u001b[1;32m 11\u001b[0m j \u001b[39m=\u001b[39m offset\n\u001b[1;32m 12\u001b[0m cm[i,j\u001b[39m-\u001b[39m\u001b[39m1\u001b[39m] \u001b[39m=\u001b[39m \u001b[39m-\u001b[39ms_x \u001b[39m*\u001b[39m alpha_interblock(alpha_x[r, j\u001b[39m-\u001b[39m\u001b[39m1\u001b[39m], alpha_x[r, j])\n\u001b[0;32m---> 13\u001b[0m cm[i,j] \u001b[39m=\u001b[39m \u001b[39m1\u001b[39m \u001b[39m+\u001b[39m s_x \u001b[39m*\u001b[39m (alpha_interblock(alpha_x[r,j], alpha_x[r, j\u001b[39m+\u001b[39;49m\u001b[39m1\u001b[39;49m]) \u001b[39m+\u001b[39m alpha_interblock(alpha_x[r, j], alpha_x[r, j\u001b[39m-\u001b[39m\u001b[39m1\u001b[39m]))\n\u001b[1;32m 14\u001b[0m cm[i,j\u001b[39m+\u001b[39m\u001b[39m1\u001b[39m] \u001b[39m=\u001b[39m \u001b[39m-\u001b[39ms_x \u001b[39m*\u001b[39m alpha_interblock(alpha_x[r, j], alpha_x[r, j\u001b[39m+\u001b[39m\u001b[39m1\u001b[39m])\n\u001b[1;32m 16\u001b[0m offset \u001b[39m+\u001b[39m\u001b[39m=\u001b[39m \u001b[39m1\u001b[39m\n", " [ 0. -1.28906303 3.7948142 -1.50575117 0. 0.\n",
"\u001b[0;31mIndexError\u001b[0m: index 5 is out of bounds for axis 1 with size 5" " 0. ]\n",
" [ 0. 0. -1.50575117 4.6735985 -2.16784733 0.\n",
" 0. ]\n",
" [ 0. 0. 0. -2.16784733 5.48692172 -2.31907439\n",
" 0. ]\n",
" [ 0. 0. 0. 0. 0. 0.\n",
" 0. ]\n",
" [ 0. 0. 0. 0. 0. 0.\n",
" 0. ]]\n"
] ]
} }
], ],