kristoferssolo/Tetris
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Tetris.git synced 2025-10-21 20:00:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(ai): get_peaks

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-05 17:42:10 +02:00
parent 1ce17c43ba
commit aca113f562
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/ai/fitness/peaks.py
View File

@ -4,14 +4,8 @@ from ai.log import log
def get_peaks(field: np.ndarray) -> np.ndarray:
col_num = field.shape[1]
peaks = np.zeros(col_num)
for col in range(col_num):
if 1 in field[:, col]:
peaks[col] = field.shape[0] - np.argmax(field[:, col], axis=0)
return peaks
peaks = np.where(field == 1, field.shape[0] - np.argmax(field, axis=0), 0)
return peaks.max(axis=0)
def get_peaks_max(field: np.ndarray) -> int: