Pythonのエラーを改善したいです

list = [
  [0,20, 10, 20, 20], 
  [0,40, 20, 40, 20], 
  [0,60, 80, 40, 90], 
  [0,140, 120, 120, ], 
]
df = pd.DataFrame(list)
df

from sklearn.linear_model import LinearRegression
def pred(df):
    x_train = df.iloc[0:df.shape[0]-1, 0:df.shape[1]-1]
    y_train = df.iloc[0:df.shape[0]-1, df.shape[1]-1]
    # x_test = [df.iloc[0:df.shape[0]-1, 0:df.shape[1]-1]]
    x_test = [df.iloc[df.shape[0]-1, 0:df.shape[1]-1]]
    model_lr = LinearRegression()
    model_lr.fit(x_train, y_train)
    return model_lr.predict(x_test)
    pred(df.T)
    #array([138.])
![イメージ説明](https://ddjkaamml8q8x.cloudfront.net/questions/2022-10-02/d637c182-cdf9-446b-a45f-11902e525c78.jpeg)

このエラーはどのようにして改善すれば良いですか？データ右下の空欄を回帰分析で求めるようにしたいです

can110

2022/10/01 21:23

dfsという変数が定義されていないようですがエラーが再現するコードを画像ではなくテキストで提示ください。

wpjrnfa

2022/10/01 23:47

回答ありがとうございます。表の左側4行(0123)の規則性を分析して右側(4)の3列目の数値を予測するようにしたいのですが、コードが間違っているのでしょうか

meg_

2022/10/02 00:54

画像を添付したかったようですが表示されていません。投稿する前にレビューで確認すると良いかと思います。

行動規範の内容に同意します

回答2件

Python
1print(df)
2#	0	1	2	3	4
3#0	0	20	10	20	20.0
4#1	0	40	20	40	20.0
5#2	0	60	80	40	90.0
6#3	0	140	120	120	NaN
7
8def pred(df):
9    x_train = df.iloc[0:df.shape[0]-1, 0:df.shape[1]-1]
10    y_train = df.iloc[0:df.shape[0]-1, df.shape[1]-1]
11    # x_test = [df.iloc[0:df.shape[0]-1, 0:df.shape[1]-1]]
12    x_test = [df.iloc[df.shape[0]-1, 0:df.shape[1]-1]]
13    model_lr = LinearRegression()
14    model_lr.fit(x_train, y_train)
15    return model_lr.predict(x_test)
16
17pred(df.T)
18#array([138.])