一つだけ要素の変わる同様の処理をループさせたい

Question

環境はpython2.7、windows10でanacondaを使用しています。
以下のようなコードで、四角柱の構造物にはたらくモーメントを高さごとに切り分けて計算したいのですが、for文でループさせる方法がわかりません。
hdfファイルには粒子法という構造物のある範囲に微小な粒子をつめこみ計算するというプログラムの計算結果がはいっていて、x,y,z座標とそれぞれの点（格子）での圧力がはいっています。

import h5py
import numpy as np
import matplotlib.pyplot as plt

#hdfファイルから位相と圧力をもってくる
hdf5 = h5py.File("file.hdf5", 'r')
list_x=(hdf5["particles/obstacle/arrays/x"].value.tolist())
list_y=(hdf5["particles/obstacle/arrays/y"].value.tolist())
list_z=(hdf5["particles/obstacle/arrays/z"].value.tolist())
list_p=(hdf5["particles/obstacle/arrays/p"].value.tolist())

#位相の条件分け
z_1 = [i for i, takasa in enumerate(list_z) if 0.020>=takasa]
z_2 = [i for i, takasa in enumerate(list_z) if 0.040>=takasa>0.020]
z_3 = [i for i, takasa in enumerate(list_z) if 0.060>=takasa>0.040]
x_1 = [i for i, x in enumerate(list_x) if 4.93>=x]
x_2 = [i for i, x in enumerate(list_x) if 5.09>=x>=5.07]

#一番下の段について前面と背面の位相をだす
zenmen_1 = set (z_1) & set (x_1)
haimen_1 = set (z_1) & set (x_2)

#それぞれの条件の位相で圧力をもってくる
p_zenmen_1 = [list_p[i] for i in zenmen_1]
p_haimen_1 = [list_p[i] for i in haimen_1]

#圧力をモーメントにする
p_goukei_1 = sum(p_zenmen_1)-sum(p_haimen_1)
mo1 = p_goukei1*0.01

#2段目について同様の処理
zenmen_2 = set (z_2) & set (x_1)
haimen_2 = set (z_2) & set (x_2)

p_zenmen_2 = [list_p[i] for i in zenmen_2]
p_haimen_2 = [list_p[i] for i in haimen_2]

p_goukei_2 = sum(p_zenmen_2)-sum(p_haimen_2)
mo2 = p_gouke2*0.03
#3段目以降省略

考える方向に圧力のかかっている前面と背面のみを抽出し、その圧力を合計し、高さごとにモーメントの式にいれたいのですが繰り返しにする方法がわかりません。稚拙な質問なのは重々承知ですが方法だけでもよいのでどなたかご教授ください。

Comments

[Lhankor_Mhy]

zyouso_1 zyouso_2 はどこかここに書かれていないコードで定義されているのですか？

[aine_]

すみません、z_1,2の写し間違いです

[Lhankor_Mhy]

とすると、z_1 z_2 z_3 は必ず存在するのですね？ そして z_4 が存在することはない、と理解してよろしいですか？

[aine_]

z_1,z_2,z_3は存在します。このコードでは省略していますがz_100くらいまで処理したいです。

[aine_]

構造物に上限は設定しているためzも上限はあります。無限に続くことはないです。

[Lhankor_Mhy]

z_n をリストにしてループを回せばいいように思うのですが、何か障害がありますか？

[aine_]

ありがとうございます。リストでできました。浅学であるため連番のファイル読み込みにつかう%(i)などと同じ処理ができるのではと勘違いしていました。

Answer 1

リストでできました。無知をさらして申し訳ない。