读取csv文档

孤者浪人 提交于 2020-02-17 15:08:20
import csv
import numpy as np
boston_dir_path = r'D:\annconda\Lib\site-packages\sklearn\datasets\data\boston_house_prices.csv'

def load_boston():
    with open(boston_dir_path) as f:
        data_file = csv.reader(f)
        temp = next(data_file)
        n_simples = int(temp[0])
        n_features = int(temp[1])
        data = np.empty((n_simples,n_features))

        target = np.empty((n_simples,))
        temp = next(data_file)
        features_name = np.array(temp)

        for i,d in enumerate(data_file):
            data[i] = np.asarray(d[:-1],dtype=np.float64)
            target[i] = np.asarray(d[-1],dtype=np.float64)
    return data,target
if __name__ == '__main__':
    data,target = load_boston()
    print(target)
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!