python rms 二乗平均平方根 求め方

Question

二乗平均平方根を求めたく下記のコードを書きましたが上手くいきません．

sp_lstは6000行252列のデータです．
最初の2000行は不要なので削除して，252列ぞれぞれの二乗平均平方根を求めたいです．
見よう見まねでhttps://teratail.com/questions/127909をもながらやっていますが
上手くいきません．どなたかご教授ください．

rms = []
another_average = pd.DataFrame(sp_lst)
#another_average = pd.read_csv('suface_average_pressure_coefficient.csv', header=None)
another_average1 = another_average.drop(range(0,2000))
another_average2 = np.array(another_average1)

for i in range(252):
	rms1 = np.sqrt((another_average1[:,i]**2).sum()/another_average1.size)

for i in range(21):
	rms_1 = (dec.Decimal(rms1[i]) + dec.Decimal(rms1[i+21]) + dec.Decimal(rms1[i+42]))/dec.Decimal(3)
	average.append(rms_1)
for i in range(21):
	rms_2 = (dec.Decimal(rms1[i+63]) + dec.Decimal(rms1[i+84]) + dec.Decimal(rms1[i+105]))/dec.Decimal(3)
	average.append(rms_2)
for i in range(21):
	rms_3 = (dec.Decimal(rms1[i+126]) + dec.Decimal(rms1[i+147]) + dec.Decimal(rms1[i+168]))/dec.Decimal(3)
	average.append(rms_3)
for i in range(21):
	rms_4 = (dec.Decimal(rms1[i+189]) + dec.Decimal(rms1[i+210]) + dec.Decimal(rms1[i+231]))/dec.Decimal(3)
	rms.append(rms_4)

np.savetxt('rms.csv',rms,delimiter=',')

Answer 1

数値データであれば、pandas でなく、最初から numpy 配列として読み込んだほうがよいです。
loadtxt の引数は こちら など見ながらファイルに合わせて適宜変更してください。

# 最初の2000行をスキップして、data.csv からデータを読み込む。
# a = np.loadtxt('data.csv', skiprows=2000, delimiter=',')

a = np.random.randn(4000, 252)  # テスト用のダミー

rms = np.sqrt(np.square(a).mean(axis=0))
print(rms.shape)  # (252,)
print(rms)

1. np.square() で自乗を計算
2. ndarray.mean(axis=0) で列ごとの平均を計算
3. np.sqrt() で平方根を計算

Comments

[Tubasa1995]

tiitoiさんのコードは1列ずつの計算コードになっていますか？
それとも，4000行252列の全データの二乗平均平方根を求めることになってますか？

[tiitoi]

列ごとということだったのですね。
コードを修正しました。