【自用】 pandas 读取 CSV 文件

2019独角兽企业重金招聘Python工程师标准>>>

import numpy as np
import pandas as pd
import csv


def load_csv(filename, target_dtype, features_dtype, target_column=-1):
    """Load dataset from CSV file with a header row."""
    pd_csv = pd.read_csv(filename)
    data_file = pd_csv.values
    data = data_file[:, 1:-1].astype(features_dtype)
    target = data_file[:, target_column].astype(target_dtype)

    return data, target

调用：

training_data, training_target = load_csv(train_file, target_dtype=np.int, eatures_dtype=np.float32)

 

    PS:自用的  不做解释                        作者：一个吃货帅锅    

转载于:https://my.oschina.net/kilosnow/blog/1619565