From 3a38b157d2369c7e56505dbbdadca7f8dc826ef4 Mon Sep 17 00:00:00 2001 From: linarphy Date: Tue, 16 May 2023 16:01:26 +0200 Subject: [PATCH] Add output --- ETA.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ETA.py b/ETA.py index 1d9da23..f6b5247 100644 --- a/ETA.py +++ b/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 )