pythpython-dataframe数据预处理（一）之txt数据集文件分割

'''
fun: 一个txt文件a有x行，y列。一行代表一个样本，第1-（y-1）代表特征，第y列代表样本的类别（即标签例如有c1/c2/c3三个类别），
此代码依据样本的标签列，将a.txt分割成 c1.txt、 c2.txt、 c3.txt。
'''
# -*- coding: utf-8 -*-
import pandas as pd
import
numpy as np
feature_size = 45
col_index = []
for i in range(feature_size):
col_index.append('f%d'%i)
col_index.append('class')
#列索引
data = pd.read_csv(r'./Tr11_end.csv',names=col_index)
rows = len(data )
#求出一共多少行
cols =
data.columns.size
# print("源文件共有 %d 行"%rows)
# print("源文件共有 %d 列"%cols)
classes = data['class'].unique()
#数据共有几个类别
for c in classes:
name = 'data' + c
data_seg = data[ data['class']==c ]
data_seg.to_csv(r'./segm'+"//" +name+".csv",index=False)
#index=false控制输出文件中不自动添加列序号
#data.columns = col_index
#重新为列索引赋值
#print (data.columns)
#列索引名称
# print (data.index)
#行索引名称
#print(data[cols-1][:10])

最后

以上就是鳗鱼皮带为你收集整理的pythpython-dataframe数据预处理（一）之txt数据集文件分割的全部内容，希望文章能够帮你解决pythpython-dataframe数据预处理（一）之txt数据集文件分割所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。