我目前有一个for循环运行大约3000次迭代,在循环结束时,我想要的值是EGr:
EG = la.eigvals(H)
EGr = EG.real
EGr是一个1 × 8的数组。例子:
[ 0.22478205 2.50936963 1.81160702 1.76320129 1.94243736 1.81346264
1.94243736 1.81346264]
在循环进行到下一次迭代之前,我如何在文本/excel文件中编写这8个值?
在循环结束时,我希望只有一个文件,其中包含循环的所有值
下面是使用csv
的基本方法:
with open("out.csv", "w") as f:
writer = csv.writer(f) #Uses Excel dialect by default
for thing in things:
#...
EG = la.eigvals(H)
EGr = EG.real
writer.writerow(EGr)