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 ):
|
for i in range( len( fall ) - 1 ):
|
||||||
stairs[ fall[ i ] : fall[ i + 1 ] , x ] = np.mean( stairs[ fall[ i ] : fall[ i + 1 ] ] )
|
stairs[ fall[ i ] : fall[ i + 1 ] , x ] = np.mean( stairs[ fall[ i ] : fall[ i + 1 ] ] )
|
||||||
stairs[ fall[ -1 ] : , x ] = 0
|
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