既然这里是TensorFlow论坛,我们就用下tensorflow吧,代码如下:
注意我只在python2下测试过
- # -*- coding: UTF-8 -*-
- import os
- import numpy as np
- import pandas as pd
- import tensorflow as tf
- def convert(filename):
- name_with_extension = os.path.basename(filename)
- name, _ = os.path.splitext(name_with_extension)
- # 以下内容为题主自己写的代码
- txt = np.loadtxt(filename)
- txt_df = pd.DataFrame(txt)
- txt_df.to_csv(name + '.csv', index=False)
- # 假设你想要批量处理的文件在某个路径
- file_dir = './test/*.txt'
- # 获得所有文件的名字列表
- file_list = tf.gfile.Glob(file_dir)
- for f in file_list:
- convert(f)
复制代码 |