机器学习笔记我的python学习道路

关于python读取csv的几种方法

2019-02-27  本文已影响1人  fred_33c7

CSV是一种常见的数据存储文件,在python里面我们应该怎么样去打开并读取它呢?

环境:python3.6 注意本文中的一些方法在python2.7中并不适用需要修改
数据csv的大概样子:


csv文件截图

1. 自己写

# f = open(path,'rt').readlines()
    f = open(path,'rt').readlines()  # f=file(path) in py2.7
    x = []
    y = []
    for i, d in enumerate(f):
        if i == 0:
            continue
        d = d.strip()
        if not d:
            continue
        print(d)
        d = map(float, d.split(','))
        d = list(d)   # 这一句话在py2.7中不需要
        x.append(d[1:-1])
        y.append(d[-1])
    pprint(x)
    pprint(y)
    print("--------------")
    x = np.array(x)
    y = np.array(y)

2. python自带库

f = open(path, 'rt').readlines()
    print(f)
    d = csv.reader(f)
    for line in d:
        print(line)

3. numpy读入

p = np.loadtxt(path, delimiter=',', skiprows=1)
    print(p)

4. pandas读入 (一般用这种)

data = pd.read_csv(path)    # TV、Radio、Newspaper、Sales
    x = data[['TV', 'Radio', 'Newspaper']]
    # x = data[['TV', 'Radio']]
    y = data['Sales']
    print(x)
    print(y)
上一篇下一篇

猜你喜欢

热点阅读