python处理数据时常用到的函数【1】

遍历目录下的所有子文件汇总到相同根目录下

首先需要介绍一个好用的函数os.wak()
功能：遍历目录下的所有文件，生成3元组 (dirpath, dirnames, filenames)【文件夹路径, 文件夹名字, 文件名】
还用到了shutil.copyfile(oldname,newname) 函数
功能：将文件从旧地址复制到新地址

import os
import shutil
for root, dirs, files in os.walk("旧地址的根目录"):
    for name in files:
        shutil.copyfile(os.path.join(root,name), "新地址"+name)

除了os.walk() 函数可以遍历文件外， os.listdir() 函数也可以列出当前文件夹下的所有文件

---

读取csv文件转换成txt文件

用到csv库，函数为csv.DictReader()

import csv
with open(***.csv) as cfile:
    reader = csv.DictReader(cfile)
    f.open("train.txt","w")
    for row in reader:
        f.write(row.get(" 列名称1")+" "+row.get(" 列名称2")+"\n")
f.close()

---

读取txt 文件中的指定列的数据

函数为 split()
功能： 以空格为标志分割文本

with open(***.txt) as file:
    for row in file:
         print(row.split()[0]) #  打印第一列的数据