python按照序號建立資料夾並且按照序號將產生的檔案填入建立的資料夾
阿新 • • 發佈:2021-01-23
技術標籤:python
我們用python建立檔案的時候需要按照一定的序號將檔案放入規定的資料夾,建立資料夾的時候需要用到索引 i,j,k 等,以下是我寫好的程式碼
import os
import numpy as np
from matplotlib import pyplot as plt
import time
path = 'G://phase8//original_npy1//'
# new_path = 'G://phase8//phs1//'
for root, dirs, files in os.walk(path):
for i in range(len(files)):
file_path = os.path.join(root, files[i])
phase = np.load(file_path)
for j in range(20):
new_path = 'G://phase8//phs1//'
if i < 10:
new_path = new_path + str('000') + str(i) + '_' + str(j)
if i >= 10 and i < 100:
new_path = new_path + str('00') + str(i) + '_' + str(j)
if i >= 100 and i < 1000:
new_path = new_path + str('0') + str(i) + '_' + str(j)
if i >= 1000:
new_path = new_path + str(i) + '_' + str(j)
os.makedirs(new_path)
phase1 = phase[256*j: 256 *j+256,:] # 長橫圖
for k in range(40):
phase2 = phase1[:, k*128:256+k*128] # 256*256的圖片 間隔為128
plt.imshow(phase2)
plt.axis('off')
plt.gcf().set_size_inches(256 / 100, 256 / 100)
plt.gca().xaxis.set_major_locator(plt.NullLocator())
plt.gca().yaxis.set_major_locator(plt.NullLocator())
plt.subplots_adjust(top=1,bottom=0,right=1,left=0,hspace=0,wspace=0)
plt.margins(0, 0)
if i < 10:
# np.save(new_path + '//' + str('000') + str(i) + str('_') + str(j) + str('_') + str(k), phase2)
plt.savefig(new_path + '//' + str('000') + str(i) + str('_') + str(j) + str('_') + str(k), dpi=100)
if i >= 10 and i < 100:
# np.save(new_path + '//' + str('00') + str(i) + str('_')+str(j) + str('_') + str(k), phase2)
plt.savefig(new_path + '//' + str('00') + str(i) + str('_') + str(j) + str('_') + str(k), dpi=100)
if i >= 100 and i < 1000:
# np.save(new_path + '//' + str('0') + str(i) + str('_')+str(j) + str('_') + str(k), phase2)
plt.savefig(new_path + '//' + str('0') + str(i) + str('_') + str(j) + str('_') + str(k), dpi=100)
if i >= 1000:
# np.save(new_path + '//' + str(i) + str('_')+str(j) + str('_') + str(k), phase2)
plt.savefig(new_path + '//' + str(i) + str('_') + str(j) + str('_') + str(k), dpi=100)
print(i, j, k)
print('---save successfully---')