import os
import numpy as np
 
def eachFile(filepath):
	j=0
	pathDir = os.listdir(filepath)  #獲取當前路徑下的文件名，返回list
	for s in pathDir:
		newDir=os.path.join(filepath,s)  #將文件名寫入到當前文件路徑后面
		if os.path.isfile(newDir): #如果是文件
		  if os.path.splitext(newDir)[1]==".txt": #判斷是否是txt
		    readFile(newDir)
		    j=j+1
		    print(j)
		    pass
		  else:
		  	break
 
 
 
def readFile(filepath):
	index=0         #控制數(shù)據(jù)存入不同的list
	with open(filepath,"r") as f:
		line=f.readline()
# 		print(line)
		while line:
			if line[:9]==' cell 10':   #根據(jù)關(guān)鍵詞抽取數(shù)據(jù)
				f.readline()
				index=index+1
				for i in range(126):    #抽取的數(shù)據(jù)格式
					energy,f4,error=f.readline().split()
					if(index==1):
						list_total.append(f4)
					elif(index==2):
						list_electron.append(f4)
					else:
						list_positron.append(f4)
			line=f.readline()
 
def main():
	global list_total,list_electron,list_positron   #定義全局變量，可以將所有數(shù)據(jù)都存入list中
	fp=r'F:\\MCwork\\MCCM\\scripts\\filesworks' #存放數(shù)據(jù)的目錄
	os.chdir(fp)
	eachFile(fp)
	output =open("flux.txt",'w')  #將list存入相應(yīng)的文件中，便于后期處理數(shù)據(jù)
	listdata_total=list(np.reshape(list_total,(15,126)).T)  #改變數(shù)組維度，存儲
	for i in range(126):           #數(shù)據(jù)讀入相應(yīng)文件的第一種方法，第一篇博客有介紹
		for j in range(15):
			output.write(listdata_total[i][j]+' ')
			output.write('\t')
		output.write('\n')
	output.close()
 
if __name__ == '__main__':
	list_electron=[]
	list_total=[]
	list_positron=[]
	main()