pythonでif文の条件分岐後の結果を足し合わせたいがうまくいかない

Question

前提

pythonでフーリエベッセル関数を用いた級数展開法を行なっています。今基底関数系のデータを決め、それぞれの基底関数系に対応する係数を求めることはできたのですが、この求めた係数を対応する基底関数系に掛けて値を足し合わせるというところで苦労しています。最終的には設定したデータと合うように像を再構成できれば良いので等高線を描くためにcontourを利用しています。

実現したいこと

if文で場合分けしている関数の値を全て足し合わせたい

発生している問題・エラーメッセージ

明らかに異なる結果が出ているのでif文でミスがあるように思われる。27~34行を変える必要があると思われる。

該当のソースコード

import numpy as np
import matplotlib.pyplot as plt
from scipy.special import jv
import pandas as pd
import itertools
import csv

csv_file = open('coeffication.csv')              #csvファイルを開く
csv_reader = csv.reader(csv_file)       #開いたcsvファイルからreaderオブジェクトを生成
clf = []                          #抽出するデータを格納する空のリストを作る

for row in csv_reader:                  #readerオブジェクトをループしてデータ抽出
    if csv_reader.line_num == 1:        #ヘッダー行はスキップする
        continue                        #Trueになる1行目はなにもしない
    clf.append(row[0])            #row行目の係数データを格納

csv_file.close()                        # csvファイルを閉じる。

clf_i = [float(s) for s in clf]  #配列は文字列型なのでint型に変換

fig = plt.figure()
ax1 = fig.add_subplot(111)
x = np.linspace(0, 4*2^(1+0), 40) #等間隔で値を０～4まで20個作成 lの値はここで変える
y = np.linspace(0, 2*np.pi, 40) #等間隔で値を０～360まで20個作成
x, y = np.meshgrid(x, y)  #ｘ,ｙからメッシュグリッドを作成

def fz(x,y):
    for e in range(41):
        if e % 2 ==0:
            a = clf_i[e]*jv(e,x)*np.cos(e*y)
        else:
            b = clf_i[e]*jv(e,x)*np.sin(e*y)
    return a+b    #足し合わせる
z = fz(x,y)

x1 = x*np.cos(y)
y1= x*np.sin(y)

ax = plt.contour(x1,y1,z,colors='black')    #等高線表示
ax.clabel(fmt='%1.1f', fontsize=16)       #等高線の値を表示
ax = plt.contourf(x1,y1,z,cmap='rainbow')   #等高線レベルに応じて色を塗る
ax = plt.colorbar(label="contour level")  #カラーバー表示


plt.show()

試したこと

対応する基底関数系の像をcontourで図示したので、結果もcontourで図示しないといけないのがネックになっている。一応3次元データをプロットすれば求めているようなグラフになるのだが、contourfの場合、高度の値に二次元を求められるためにこのような冗長な式になっている。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

27~34行を変える必要があると思われる。

fz 関数は以下の様になるかと思います。

def fz(x,y):
    z = np.zeros(x.shape)
    for e in range(41):
        if e % 2:
            z += clf_i[e]*jv(e,x)*np.sin(e*y)
        else:
            z += clf_i[e]*jv(e,x)*np.cos(e*y)
    return z

Comments

[NMKN]

np.zeros(x.shape)は知りませんでした。ありがとうございます！参考にさせていただきます！

Answer 2

ぱっと見た感じfz関数に問題があります。
ループの最後(e==40)のaとbだけがreturn a+bされています。(e=0〜38のaとbは使われていない)