机器学习:Python实践
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人

5.4 采用NumPy导入数据

也可以使用NumPy的loadtxt()函数导入数据。使用这个函数处理的数据没有文件头,并且所有的数据结构是一样的,也就是说,数据类型是一样的。代码如下:

    from numpy import loadtxt
    # 使用NumPy导入CSV数据
    filename='pima_data.csv'
    with open(filename, 'rt')as raw_data:
        data=loadtxt(raw_data, delimiter=',')
        print(data.shape)

这段代码就比直接使用Python的标准类库简洁了很多,执行结果如下:

   (768, 9)