83 lines
1.9 KiB
Python
83 lines
1.9 KiB
Python
import os.path
|
|
import sys
|
|
import glob
|
|
import numpy as np
|
|
import math
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def Import_dessinBitmap(file):
|
|
array = np.load(file)
|
|
return array.tolist()
|
|
|
|
|
|
def Afficher_array(y):
|
|
space = ' ' # <--- Pour modifier les espaces
|
|
N1 = len(y)
|
|
for i in range(N1):
|
|
N2 = len(y[i])
|
|
for j in range(N2):
|
|
if y[i][j]:
|
|
print(str(y[i][j]), end=space)
|
|
else:
|
|
print(' ', end=space)
|
|
print("")
|
|
|
|
|
|
def Afficher_as_txt(y):
|
|
space = ' ' # <--- Pour modifier les espaces
|
|
plein = '#'
|
|
vide = ' '
|
|
|
|
N1 = len(y)
|
|
for i in range(N1):
|
|
N2 = len(y[i])
|
|
for j in range(N2):
|
|
if y[i][j]:
|
|
print(plein, end=space)
|
|
else:
|
|
print(vide, end=space)
|
|
print("")
|
|
|
|
|
|
def Print_plt(Mat, fichier, rep=".\\", afficher=False, save=True):
|
|
x = []
|
|
y = []
|
|
c = "blue" # <--- Couleur des icones
|
|
mk = "o" # <--- Motifs des icones ("o", "x", "1")
|
|
size = 30 # <--- Taille des icones
|
|
N1 = len(Mat)
|
|
for i in range(1, N1 - 1):
|
|
N2 = len(Mat[i])
|
|
for j in range(1, N2-1):
|
|
if Mat[i][j]:
|
|
x.append(j)
|
|
y.append(N1-i)
|
|
plt.scatter(x, y, color=c, s=size, marker=mk)
|
|
plt.title(fichier)
|
|
plt.axis([0, N1, 0, N2])
|
|
if save:
|
|
plt.savefig(rep+fichier+".png")
|
|
if afficher:
|
|
plt.show()
|
|
plt.close()
|
|
|
|
|
|
def Print_mat(y, fichier, rep=".\\", afficher=False, save=True):
|
|
heatmap = plt.pcolor(y)
|
|
plt.colorbar(heatmap)
|
|
if save:
|
|
plt.savefig(rep+fichier+".png")
|
|
if afficher:
|
|
plt.show()
|
|
plt.close()
|
|
|
|
|
|
def main(repc=".\\", reps=".\\"):
|
|
for f in glob.glob(repc + "*.npy"):
|
|
y = Import_dessinBitmap(f)
|
|
Print_mat(y, f, rep=reps)
|
|
|
|
|
|
main()
|