前提・実現したいこと
ここに質問の内容を詳しく書いてください。
Pythonで株価予測と共に日付を予測するコードを作成しております。
下記エラーが出て、解決ができません。
読み込んだcsvのyear列をindexにして、データ日付列、終値列をそれぞれトレーニングデータとテストデータにしたいと考えております。冒頭のコードのみでエラーがでましたので、後段のコードは省略致します。
発生している問題・エラーメッセージ
ValueError: Data must be 1-dimensional
該当のソースコード
Python
1from sklearn.model_selection import train_test_split 2import pandas as pd 3 4table = pd.read_csv("table.csv",index_col='year') 5 6X = table.values[:, 0] 7X = pd.DataFrame(map(lambda x: x.replace(",",""),X)).astype('float').values 8y = table.values[:, 1] 9 10X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0, shuffle=False) 11X_train_std = (X_train - X_train.mean()) / X_train.std() 12X_test_std = (X_test - X_train.mean()) / X_train.std() 13 14# df_trainというテーブルを作りそこにindexをyear、カラム名をデータ日付、終値にしてdf_train.csvという名前でdataフォルダ内に出力する。 15df_train = pd.DataFrame( 16 {'データ日付': X_train_std, 17 '終値': y_train}, 18 columns=['データ日付', '終値'], 19 index=table.index[:len(X_train_std)]) 20df_train.to_csv('./df_train.csv')
試したこと
Xは日付データを引用しているため、他の数値データなどに変更して試しましたが、同じエラーが出ました。
補足情報(FW/ツールのバージョンなど)
特になし
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。