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

feat(ai): add aggregate_height

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-09 20:28:31 +02:00
parent 89884d7e2a
commit 9daa4f6509
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/ai/heuristics/height.py
View File

@ -2,4 +2,13 @@ import numpy as np
def aggregate_height(field: np.ndarray[int, np.dtype[np.uint8]]) -> int: def aggregate_height(field: np.ndarray[int, np.dtype[np.uint8]]) -> int:
return 0 """
Calculates the aggregate height of the field.
Args:
field: 2D array representing the game field.
Returns:
The aggregate height of the field.
"""
return int(np.sum(field.shape[0] - np.argmax(field, axis=0)))