From c943b931239e3ea4744ce19190019ae05b57d029 Mon Sep 17 00:00:00 2001 From: juju12345 <14647639+juju12345@user.noreply.gitee.com> Date: Fri, 5 Jul 2024 19:38:23 +0800 Subject: [PATCH] =?UTF-8?q?=E7=8E=8B=E6=B6=B5=E7=9A=84=E4=BD=9C=E4=B8=9A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- datas.py | 157 +++++++++++++++++++++++++++++++++++++++++++++++++++++ predict.py | 18 ++++++ 2 files changed, 175 insertions(+) create mode 100644 datas.py create mode 100644 predict.py diff --git a/datas.py b/datas.py new file mode 100644 index 0000000..704e542 --- /dev/null +++ b/datas.py @@ -0,0 +1,157 @@ +# 模拟太原和吕梁的房价 +# 城市x1 面积x2 户型x3 是否学区房x4 (1 2) 装修的风格x5 (1 2) +datas = [ + { + "city": "吕梁", + "area": 100, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8000, + }, + { + "city": "吕梁", + "area": 120, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8300, + }, + { + "city": "吕梁", + "area": 130, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8600, + }, + { + "city": "吕梁", + "area": 140, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9000, + }, + { + "city": "吕梁", + "area": 150, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9300, + }, + { + "city": "吕梁", + "area": 160, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9500, + }, + { + "city": "吕梁", + "area": 170, + "rooms": 4, + "school": 3, + "style": 2, + "price": 9500, + }, + { + "city": "吕梁", + "area": 180, + "rooms": 4, + "school": 4, + "style": 3, + "price": 9600, + }, + { + "city": "吕梁", + "area": 190, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9400, + }, + { + "city": "太原", + "area": 210, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9200, + }, + { + "city": "太原", + "area": 200, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9100, + }, + { + "city": "太原", + "area": 190, + "rooms": 2, + "school": 1, + "style": 1, + "price": 9000, + }, + { + "city": "吕梁", + "area": 180, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8900, + }, + { + "city": "太原", + "area": 170, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8800, + }, + { + "city": "太原", + "area": 160, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8700, + }, +] + + +import numpy as np +from datas import datas +X=[] +Y=[] +cityMark={"吕梁":1,"太原":2} +for item in datas: + single=[] + +single.append(cityMark[item["city"]]) +single.append(item["rooms"]) +single.append(item["school"]) +single.append(item["style"]) +single.append(item["area"]) +X.append(single) +Y.append(item["price"]) + +X=np.array(X) +Y=np.array(Y) + +theta=np.linalg.pinv(X.T.dot(X)).dot(Y) + + +print(theta) + +print(theta.dot(np.array([2,100,2,1,1]))) + + + + + + diff --git a/predict.py b/predict.py new file mode 100644 index 0000000..67c19c6 --- /dev/null +++ b/predict.py @@ -0,0 +1,18 @@ +# 1.先安装一个 科学计算的框架 +# import numpy as np +#a=np.array([[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]]) +#b=np.array([1,2,3,4,5]) +#求点乘 +#print(a.dot(b)) +#print(a) +# 奇异矩阵 print(np.linalg.inv(a)) +# 排除掉奇异矩阵求幂 +#print(np.linalg.pinv(a)) +import numpy as np +a=np.array([[3,2,3],[3,2,3],[3,2,3]]) +b=np.array([1,2,3]) + + +print(a.T) +print(a.dot(b)) +print(np.linalg.pinv(a)) -- Gitee