Pythonで標準化のプログラムを作成しています。

Question

前提・実現したいこと

Python初心者です。
標準化のプログラムを作成しています。
１．ファイル名を標準入力で受け取り
２．ファイルの各列を標準化
３．結果を新しいcsvファイルに出力
といった流れを考えています。
エラーの意味が分からず、解決策をご提示いただけないでしょうか。
何卒、よろしくお願いいたします。

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

Traceback (most recent call last):
  File "std_new.py", line 19, in <module>
    x = data[:,i]
IndexError: index 4 is out of bounds for axis 1 with size 4

該当のソースコード

import numpy as np
import csv

def standardize(x):
    x_mean = x.mean()
    std = x.std()

    return(x - x_mean)/std

filename = input("ファイル名を入力してください：")

data = np.loadtxt(fname=filename,
                  dtype = 'double',
                  delimiter = ',',
                  skiprows = 1
                  )

for i in range(len(data)):
    x = data[:,i]
    standardize_x = standardize(x)

standardize_x = csv.writer('std_result.csv')

Answer 1

列数を取得したい場合、以下のようにするべきではないでしょうか。

- for i in range(len(data)):
+ for i in range(data.shape[1]):

追記

import numpy as np
import csv


def standardize(x):
    mean = x.mean()
    std = x.std()

    return (x - mean) / std


data = np.loadtxt("sample.csv", delimiter=",", skiprows=1)


normalized = np.empty_like(data)

for i in range(data.shape[1]):
    normalized[:, i] = standardize(data[:, i])

np.savetxt("normalized.csv", normalized, delimiter=",")

Comments

[peyang]

ご回答ありがとうございます。
shapeに変更し、書き込みのところをsavetxtに変更したら正しく実行されました。
但し、入力のcsvが4列であったのに対して、出力結果が最後の列のみでした。
for文の中で上書きしてしまっているのだと思います。
別の列にそれぞれ格納できるよう改良を加える必要があるようです。
少しあがいてみます。

[tiitoi]

normalized = np.empty_like(data) で data と同じ大きさの配列を作って、標準化した列の結果を同じ列に代入すればいいかと思います。
コードを追記しました。

[peyang]

納得いきました。コードのご提示までしていただきありがとうございます。
この方法ならfor文の中で順次データを配列に格納することができました。
大変勉強になりました。