1. 程式人生 > >使用python讀取csv並轉化成list of list的方法

使用python讀取csv並轉化成list of list的方法

很多面試題目都需要處理csv檔案,以下是幾種種處理csv檔案的方法:

方法一: 使用Python基礎語言編寫

def read_csv(file_name):
    f = open(file_name, 'r')
    content = f.read()
    final_list = list()
    rows = content.split('\n')
    for row in rows:
        final_list.append(row.split(','))
    return final_list

方法二: 使用CSV 模組

import csv 

f = open(file_name, 'r')
csvreader = csv.reader(f)
final_list = list(csvreader)

方法三:使用numpy模組

import numpy as np 

nd = np.genfromtxt(filename, delimiter=',', skip_header=True)

final_list = nd.tolist()

方法四:使用pandas模組

import pandas as pd 

df = pd.read_csv(filename, index_col=0)

final_list = df.tolist()