
用numpy来批量读取处理txt文本,并批量保存。
import os
import numpy as np
from mhb_function import mhb_delete
path = "E:\Pointnet++\pointnet2_pytorch-master\data\shapenetcore_partanno_segmentation_benchmark_v0_normal/03467517" #文件夹目录
files= os.listdir(path) #得到文件夹下的所有文件名称
#输出该文件夹下所有的文件名称
print(files)
#文件总的数量
files_len = len(files)
print(files_len)
#定义一个空数组
data = {}
#利用循环依次将文件的数据存储到data这个大数组
for i in range(len(files)):
data[i] = np.loadtxt('E:\Pointnet++\pointnet2_pytorch-master\data\shapenetcore_partanno_segmentation_benchmark_v0_normal/03467517/'+files[i])
#对读取的数据进行一定的修改(用自己的函数)
rm_data = {}
# print(data[0])
for i in range(len(files)):
rm_data[i] = mhb_delete(data[i])
#将文件依次保存到文件夹中,默认所在的文件夹
for i in range(len(files)):
txt_data = np.savetxt('{}.txt'.format(i), rm_data[i])
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)