tensorflow 数据的读取与预处理

# -*- coding: utf-8 -*-
"""
Created on Thu Feb  2 07:19:22 2023

@author: Dell
"""
import os
import pandas as pd
import tensorflow as tf

os.makedirs(os.path.join('..','data'), exist_ok=True)                          # 在当前的目录下创建一个文件夹
data_file = os.path.join('..','data','house_tiny.csv')                         # 生成一个文件路径

# 这里 with 的用法可以理解为，执行后面这块代码之前捎带着把 open(data_file) as f 先执行一下
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')
    
data = pd.read_csv(data_file)
print(data)

data = pd.get_dummies(data, dummy_na=True)
print('转换data中各元素的数据类型\n',data)

# 将dataframe格式的数据转换为tensorflow的张量格式
data_values = data.values
print('获取data的值\n',data_values)
data_t = tf.constant(data_values)
print('将获取的值转换为 tensorflow 的张量格式\n',data_t)

# 前面写入数据的时候，写入空格与没有写入的效果是不同的，对后续处理的影响也是不一样的
data = data.fillna(data.mean())
print(data)