Add output

This commit is contained in:
linarphy 2023-05-16 16:01:26 +02:00
parent 836b453d8b
commit 3a38b157d2
No known key found for this signature in database
GPG key ID: 3D4AAAC3AD16E79C

7
ETA.py
View file

@ -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 )