前提・実現したいこと
pythonにおいて、N分木探索を再帰関数によって作ったとき、末端ノードに関数が到達した回数をglobal
変数によってカウントしたいのですが、思い道理にカウントできないため、質問いたしました。
お手数ですが、もしわかる方がいらっしゃいましたら、ご回答お願いいたします。
発生している問題
以下の関数は、2分木探索を再帰的に行い、2**N(Nは任意の実数)の要素数のfloat型配列a
を0
,1
の値に変換するプログラムです。以下の再帰関数の最初のif
文において、探索木の葉ノードの値を出力するのですが、その値を関数EST_codeword
に格納します。しかし、t
のインクリメントがうまくされず、EST_codeword
に変数が格納されません。
###該当のソースコード
numpy
1#2**N(Nは任意の実数)の要素数の配列``a``を入力し、値を2分木探索して計算する関数 2 3#node operation 4 5EST_codeword=np.zeros(len(a)) 6t=0 7def SC_decoding(a): 8 global t 9 #leaf node operation 10 if a.shape[0]==1: 11 if a>=0: 12 tmp=np.zeros(1) 13 elif a<0: 14 tmp=np.ones(1) 15 EST_codeword[t]=tmp #葉ノードが返す値を格納する関数 16 print(t) #ここでtが正しくインクリメントされているかどうかを確認する 17 t=+1 18 return tmp 19 20 #interior node operation 21 #step1 left input a output u1_hat #二分木の左側の計算 22 tmp1=np.split(a,2) 23 f_half_a=np.sign(tmp1[0])*np.sign(tmp1[1])*np.amin(np.abs(a)) 24 u1=SC_decoding(f_half_a) 25 26 #step2 right input a,u1_hat output u2_hat #二分木の右側の計算 27 tmp2=np.split(a,2) 28 g_half_a=tmp2[1]+(1-2*u1)*tmp2[0] 29 u2=SC_decoding(g_half_a) 30 31 #step3 up input u1,u2 output a_hat #入力されたaを計算した値a_hatを求める 32 res=np.concatenate([(u1+u2)%2,u2]) 33 return res 34
発生している問題・エラーメッセージ
試しに、
a=[-0.80082963 1.3845885 -0.90126979 0.69719649 -1.31832405 -0.1590969 1.55812294 -0.494428 -0.39330792 -0.65102752 0.86097633 1.00317875 1.04088517 0.64219232 0.61946126 -1.10920215]
について、上記の関数を適応させた結果を、出力すると、
python
10 21 31 #(t=)2となる予想だった 41 #3 51 #4 61 #5,.... 71 81 91 101 111 121 131 141 151 161 #ここまで、tの値 17[0. 1. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] #関数の出力した``EST_codeword``の値
となってしまい、t
の値が1以上になっていないことがわかります。(本当はプリントされるたびにインクリメントされていてほしい)
なお、関数の出力であるa_hat
の値に誤りはありませんでした。
補足情報(FW/ツールのバージョンなど)
google colabにて記述
参考にしたwebサイト
https://www.youtube.com/watch?v=O3JWkvEY8Lc (英語ですが、スライドを用いてアルゴリズムの説明をしています。)
回答1件
あなたの回答
tips
プレビュー