From c15aa6b17a1262573cfc89e290d5179c5f914585 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9E=97=E6=9E=97?= <6969846@qq.com> Date: Wed, 12 Mar 2025 17:26:04 +0800 Subject: [PATCH] =?UTF-8?q?=E7=AC=AC=E4=B8=80=E6=AC=A1=E6=8E=A8=E9=80=81?= =?UTF-8?q?=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Preliminary knowledge/Data_Operation.py | 27 +++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 Preliminary knowledge/Data_Operation.py diff --git a/Preliminary knowledge/Data_Operation.py b/Preliminary knowledge/Data_Operation.py new file mode 100644 index 0000000..96a340f --- /dev/null +++ b/Preliminary knowledge/Data_Operation.py @@ -0,0 +1,27 @@ +import torch + +# 张量 +import os + +os.makedirs(os.path.join('..', 'data'), exist_ok=True) +data_file = os.path.join('..', 'data', 'house_tiny.csv') +with open(data_file, 'w') as f: + f.write('NumRooms,Alley,Price\n') # 列名 + f.write('NA,Pave,127500\n') # 每行表示一个数据样本 + f.write('2,NA,106000\n') + f.write('4,NA,178100\n') + f.write('NA,NA,140000\n') + +import pandas as pd + +data = pd.read_csv(data_file) + + +inputs, outputs = data.iloc[:, 0:1], data.iloc[:, 2] +inputs = inputs.fillna(inputs.mean()) + +mid = data.iloc[:, 1:2] + +mid = pd.get_dummies(mid, dummy_na=True) + +print(mid) \ No newline at end of file