【Python】读取csv文件的方法

csv文件类型

CSV是一种以逗号分隔数值的文件类型，在数据库或电子表格中，常见的导入导出文件格式就是CSV格式，CSV格式存储数据通常以纯文本的方式存数数据表。

我们所用文件内容如下，第一行为标题行。

 

                                                  

自带库读入

 

#!/usr/bin/python
# -*- coding:utf-8 -*-

import csv

if __name__ == "__main__":

    path = 'Advertising.csv' #  默认目录
    
    #Python自带库
    f = open(path, 'r')
    print(f) #打印
    d = csv.reader(f)
    # 逐行读入
    for line in d:
        print(line)
    f.close()

numpy读入

#!/usr/bin/python
# -*- coding:utf-8 -*-

import numpy as np

if __name__ == "__main__":

    path = 'Advertising.csv'

    # delimiter:以逗号分隔  skiprows:跳过前1行
    p = np.loadtxt(path, delimiter=',', skiprows=1)
    print(p)

pandas读入

#!/usr/bin/python
# -*- coding:utf-8 -*-

import pandas as pd

if __name__ == "__main__":

    path = 'Advertising.csv'

    # pandas读入
    data = pd.read_csv(path)    # TV、Radio、Newspaper、Sales
    print(data)
    
    # 按标题分离数据
    x = data[['TV', 'Radio', 'Newspaper']]
    y = data['Sales']
    print(x)
    print(y)