最小二乗法を使った近似直線の書き方

Question

前提・実現したいこと

近似直線を描く
岩手県と秋田県の人口と死亡者数の相関を調べるためのグラフを製作しており、近似直線を書きたいのですが、近似直線が1つしか書けません。どのようにすればいいですか。

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

近似直線が書けないことです。
エラーメッセージは特にありません
下の画像は岩手県と秋田県の人口と死亡者数のデータです。1列目は岩手県の人口、2列目は岩手県の死亡者数、3列目は秋田県の人口、4列目は秋田県の死亡者数です。

コード
import numpy as np
import matplotlib.pyplot as plt
# データの読み込み
data = np.loadtxt("aaaaaa.csv", delimiter=",",usecols=[0,1,2,3])
x1 = data[: , 0] # population of Iwate
x2 = data[: , 2]#population of Akita
y1= data[: , 1] # 死亡数 of Iwate
y2= data[: , 3] # 死亡数 of Akita
#図１
plt.figure(figsize=[6,6])
plt.xlabel('population')
plt.ylabel('死亡数')
plt.scatter(x1,y1,c="red")
plt.scatter(x2,y2,c="blue")

print("iwate…red")
print("akita…blue")

# 最小二乗法
x1_mean = np.mean(x1)
x2_mean = np.mean(x2)
y1_mean = np.mean(y1)
y2_mean = np.mean(y2)
x1_var = np.var(x1)
x2_var = np.var(x2)

x1y1_cov = np.cov(x1,y1,bias=True)[0,1]
x2y2_cov = np.cov(x2,y2,bias=True)[0,1]
# 回帰係数
A1 =x1y1_cov / x1_var
B1= y1_mean - A1*x1_mean
A2 =x2y2_cov / x2_var
B2= y2_mean - A2*x2_mean
# 回帰直線の関数

def reg(x1) :
   y1= A1*x1 + B1
   return y1
def reg(x2) :
 y2= A2*x2 + B2
 return y2

# プロット用x,y
x1p = [950000,1400000] # xの範囲（任意）
y1p = [reg(x1p[0]),reg(x1p[1])]
x2p = [950000,1400000] # xの範囲（任意）
y2p = [reg(x2p[0]),reg(x2p[1])]

# 図化
plt.figure(figsize=[6,6])
plt.xlim(950000,1400000)
plt.ylim(10000,20000)
plt.xlabel('population')
plt.ylabel('死亡数')
plt.scatter(x1,y1)
plt.scatter(x2,y2)
plt.plot(x1p,y1p,color="red")
plt.plot(x2p,y2p,color="blue")
plt.show()

試したこと

ソースコードを作り直すなど

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

原因は同名の関数を定義していることです。

def reg(x1) :
   y1= A1*x1 + B1
   return y1

def reg(x2) :
 y2= A2*x2 + B2
 return y2

やっつけの修正ですが、関数名を reg1() と reg2() に変更します。

def reg1(x1) :
   y1= A1*x1 + B1
   return y1

def reg2(x2) :
 y2= A2*x2 + B2
 return y2

# プロット用x,y
x1p = [950000,1400000] # xの範囲（任意）
y1p = [reg1(x1p[0]),reg1(x1p[1])]
x2p = [950000,1400000] # xの範囲（任意）
y2p = [reg2(x2p[0]),reg2(x2p[1])]