python之数据集制作读取——CSV格式
在Python机器学习中，如何读取CSV数据呢，可以通过cvs或者pandas两种⽅式读取。
def Load_csv(csv_file_name="test.csv"):
"""
从CSV⽂件中读取数据信息
:param csv_file_name: CSV⽂件名
:return: Data：⼆维数组
"""
import csv
csv_reader = ader(open(csv_file_name))
Data=[]
for row in csv_reader:
Data.append(row)
print("Read All!")
return Data
def get_mnist_test_data():
"""label 在前，相应的参数可以根据实际需求更改"""
import pandas as pd
import numpy as np
data = pd.read_csv('Train.csv')
images = data.iloc[:,1:].values
labels = data.iloc[:,:1].values.ravel()
images = np.multiply(images,1.0/255.0)
images = shape(images.shape[0],1,28,28)
# print(images)
return images,labels
def get_mnist_test_data1():
"""label在后，相应的参数可以根据实际需求更改"""
import pandas as pd
import numpy as np
data = pd.read_csv('Train.csv')
images = data.iloc[:,0:784].values
labels = data.iloc[:,784:785].values.ravel()
images = np.multiply(images,1.0/255.0)
images = shape(images.shape[0],1,28,28)
return images,labels
如何将⽂件中的图像数据存储成CSV⽂件呢？通过数据把图像数据.append到数组中，然后通过CSV将数组中的数据下⼊CSV⽂件。
def Write_csv(Data=[[None]],csv_file_name="Train.csv"):
"""
向CSV⽂件中写⼊数据信息，有标签的话存放在数据最后位
:param csv_file_name: CSV⽂件名
:return: None，⽂件夹中会显⽰写⼊的数据信息
"""
import csv
with open(csv_file_name,"w",newline='') as csvfile:
writer = csv.writer(csvfile)#写出来存在逗号
# 没有逗号的写⼊⽅式，但是在读取的时候需要更改并处理数据
# writer=csv.writer(csvfile, delimiter=' ',quotechar='|', quoting=csv.QUOTE_MINIMAL)        #写⼊多⾏⽤writerows，写⼊单⾏⽤writerow
writer.writerows(Data)
print("Write All!")
def Write_Data():
import os
import cv2
#第⼀层⽂件夹⽬录
filename="..\\imgout\\"
Data = []
for file_1 in os.listdir(filename):
#第⼆层⽂件夹⽬录，第⼀层⽂件夹下的下⼀层⽂件夹⽬录
# print(file_1)file_1为⽂件名
filename_1 = os.path.join(filename, file_1)
Data_tmp = []
# print(file_1)
for file_2 in os.listdir(filename_1):
#第三层⽂件夹，到对应⽂件的地址
filename_2 = os.path.join(filename_1, file_2)
Data_tmp1=[]
# print(filename_2)
img = cv2.imread(filename_2,0)
python怎么读csv数据w,h=img.shape[:2]
# shape(img,[w*h])
for i in range(w):
for j in range(h):
Data_tmp1.append(img[i,j])
#添加标签
Data_tmp1.append(file_1)
Data_tmp.append(Data_tmp1)
Data=Data+Data_tmp
Write_csv(Data)

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。