【自用】 pandas 读取 CSV 文件

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:自用的  不做解释                        作者:一个吃货帅锅    

猜你喜欢

转载自my.oschina.net/kilosnow/blog/1619565