|
|
发表于 2018-5-11 11:06:02
|
显示全部楼层
- import os
- def createdirs():
- # create dirs
- workpath = os.getcwd()
- for i in range(3):
- dirname = '%s%s'%('dir', str(i))
-
- os.mkdir(dirname)
- os.chdir(dirname)
- for j in range(3):
- filename = '%s%s'%('file', str(j))
- with open(filename, 'w', encoding='utf-8') as f:
- f.write('%s%s'%(dirname, filename))
- os.chdir(os.pardir)
-
- def main():
- createdirs()
- line_list = []
- dir_list = os.listdir()
-
- for item in dir_list:
- if os.path.isdir(item):
- os.chdir(item)
-
- file_list = os.listdir()
-
- for file in file_list:
- if os.path.isfile(file):
- with open(file, 'r', encoding='utf-8') as f:
- line = f.readline()
- line_list.append(line)
- os.chdir(os.pardir)
- break
-
- with open('new_file.txt', 'w', encoding='utf-8') as new_file:
- for each_line in line_list:
- new_file.write(each_line )
-
-
- if __name__ == '__main__':
- main()
复制代码
写的不算严谨 勉强算是实现 |
|