ファイルで読み込んだデータを使ってnp.polyfitをしたい

Question

実現したいこと

2つのデータの回帰直線の傾きと切片が知りたいです。
「該当のソースコード」の13行目までは正しく実行でき、正しい解が得られています。
参考にしたページはこちらです。

入力をファイル読み込みにしたいと思い、
17、18行目で2つのファイルを読み込み、np.polyfitで書きたいと思いましたが、
できなかったのでどなたか教えてください。

weight.csvの中身は以下の通りです。
60
55
80
65

height.csvの中身は以下の通りです。
170
165
182
172

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

Traceback (most recent call last):
  File "〇〇〇\testCV338.py", line 19, in <module>
    fit = np.polyfit(x, y, 1)
          ^^^^^^^^^^^^^^^^^^^
  File "〇〇〇\numpy\lib\polynomial.py", line 637, in polyfit
    raise TypeError("expected 1D vector for x")
TypeError: expected 1D vector for x

該当のソースコード

import pandas as pd
import numpy as np
import csv

x = np.array([60, 55, 80, 65])
y = np.array([170, 165, 182, 172])

n=len(x)
t_xy=sum(x*y)-(1/n)*sum(x)*sum(y)
t_xx=sum(x**2)-(1/n)*sum(x)**2
slope=t_xy/t_xx
intercept=(1/n)*sum(y)-(1/n)*slope*sum(x)
print('傾き=',slope,'切片=',intercept)#傾き= 0.6571428571428571 切片= 129.53571428571428　Excelでも確認済

##############################
x = pd.read_csv('weight.csv', header=None)#60, 55, 80, 65
y = pd.read_csv('height.csv', header=None)#170, 165, 182, 172

fit = np.polyfit(x, y, 1)
print('[傾き,切片]=', fit)

試したこと

入力ファイルが列ベクトルになっていることが問題と思いましたが、
どのように解決するかが分かりませんでした。

Answer 1

TypeError: expected 1D vector for x

なので、np.polyfit() に 1D vector を渡してあげればよいかと思います。

x = pd.read_csv('weight.csv', header=None) #60, 55, 80, 65
y = pd.read_csv('height.csv', header=None) #170, 165, 182, 172

fit = np.polyfit(x[0], y[0], 1)
print('[傾き,切片]=', fit)

# 傾き= 0.6571428571428571 切片= 129.53571428571428
# [傾き,切片]= [  0.65714286 129.53571429]

Comments

[yyicp]

できました。
どうもありがとうございました。