您的位置:首页 > 其它

【自用】 pandas 读取 CSV 文件

2018-02-07 00:00 253 查看
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:自用的 不做解释 作者:一个吃货帅锅

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  pandas CSV