Add output
This commit is contained in:
parent
836b453d8b
commit
3a38b157d2
1 changed files with 5 additions and 2 deletions
7
ETA.py
7
ETA.py
|
@ -402,5 +402,8 @@ for x in range( size ):
|
|||
for i in range( len( fall ) - 1 ):
|
||||
stairs[ fall[ i ] : fall[ i + 1 ] , x ] = np.mean( stairs[ fall[ i ] : fall[ i + 1 ] ] )
|
||||
stairs[ fall[ -1 ] : , x ] = 0
|
||||
plt.imshow( stairs )
|
||||
plt.show()
|
||||
|
||||
if output == None:
|
||||
print( stairs )
|
||||
else:
|
||||
np.savetxt( output , stairs )
|
||||
|
|
Loading…
Reference in a new issue