python導入csv/txt文件

 1. 導入csv文件

### python導入csv文件的三種方法
 ```python #原始的方式
 lines = [line.split(',') for line in open('iris.csv')] df = [[float(x) for x in line[:4]] for line in lines[1:]] #使用numpy包

import numpy as np lines = np.loadtxt('iris.csv',delimiter=',',dtype='str') df = lines[1:,:4].astype('float') #使用pandas包

import pandas as pd df = pd.read_csv('iris.csv') df=df.ix[:,:4]

這三種方法中最後一種最簡單，不過花費時間比較長一點，第一種最麻煩，不過用時最短。這個能夠經過ipython中的magic函數%%timeit來看。

 

2. 導入txt文件

數據：

將數據轉爲DataFrame格式，代碼以下：

def read_file(path,is_test=False):
    fp = open(path,encoding='utf-8')
    dataset = []
    for line in fp.readlines():
        line = line.strip().split('\t')
        if is_test:
            line.append('-1')
        dataset.append(line)
    data = pd.DataFrame(dataset)
    data.columns = ['prefix','query_prediction','title','tag','label']
    return data

　　

 

 

 

 

參考文獻：

【1】numpy.loadtxt()