python中各种格式的文件读取和保存(csv, txt, xlsx, 图片)

本文用于学习python中各种文件格式的文件读取和保存(csv, txt, xlsx, 图片)！！！

#! _*_ coding='UTF-8' _*_
import numpy as np
import pandas as pd
import csv
import os
from PIL import Image

#读取xlsx格式的文件 第一个sheet,读取行索引和列索引  DataFrame格式
data_set = pd.read_excel('filename.xlsx', sheet_name=0, header=None)
#将DataFrame格式转化为array格式
data_set = np.array(data_set, dtype=float)
print("数据类型：",type(data_set), data_set)


#读取csv格式的文件
data_set = pd.read_csv('filename.csv', header=None)
#将DataFrame格式转化为array格式
data_set = np.array(data_set)
print("数据类型：",type(data_set), data_set)


#保存文件为csv格式  需首先将数据转化为DataFrame格式
data_set = pd.DataFrame(data_set)
data_set.to_csv('save_filename.csv')


#使用csv模块读取csv文件 'r'表示读模式
with open('filename.csv', 'r') as file:
    data_set = csv.reader( file )
    print(type(data_set))
    temp_set = []
    for line in data_set:
        temp_set.append(line)


#使用csv模块保存为csv格式 'w'表示写模式
with open('saved_filename.csv', 'w') as file:
    data_set = csv.writer( file )
    # data_set.writerows( temp_set ) #一次保存完
    for i in range(len(temp_set)): #分开保存
        data_set.writerow( temp_set[i] )


#读取图片文件夹下的图片
root_path = 'D:/Code_py/Python库/'
images_project = 'image_filesname/'
images_path = os.path.join(root_path, images_project)
#将图片名称存入列表
images_list = os.listdir(images_path)
print(images_list)
for image_path in images_list:
    with Image.open(os.path.join(images_path, image_path)) as im:
        #打印图片的信息
        # print(im)
        #保存图片为另一种格式
        im.save( os.path.join(images_path, image_path[:-4]+'.png') )


#读取txt文件夹下的txt文件
root_path = 'D:/Code_py/Python库/'
txts_project = 'txt_filesname/'
txts_path = os.path.join(root_path, txts_project)
#将图片名称存入列表
txts_list = os.listdir(txts_path)
print(txts_list)
all_txt = []
for txt_path in txts_list:
    with open(os.path.join(txts_path, txt_path),'r',encoding='UTF-8') as txt:
        #读取当前txt文件的所有内容
        txt_lines = txt.readlines()
    txt_label = []
    #将当前txt文件的每行切割
    for line in txt_lines:
        label = line.strip().split(',')
        txt_label.append(label)
    #保存txt文件为另一种格式
    np.save( os.path.join(txts_path, txt_path[:-4]+'.npy'), txt_label )
    #存储所有的txt文件内容
    all_txt.append(txt_label)