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

fix(ai): aggregate_height function

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-11 18:59:51 +02:00
parent 18ec2ba562
commit b2bea7d03a
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -11,4 +11,10 @@ def aggregate_height(field: np.ndarray[int, np.dtype[np.uint8]]) -> int:
Returns: Returns:
The aggregate height of the field. The aggregate height of the field.
""" """
return int(np.sum(field.shape[0] - np.argmax(field, axis=0))) heights = np.zeros(field.shape[1], dtype=np.uint8)
for col in range(field.shape[1]):
for row in range(field.shape[0]):
if field[row, col] != 0:
heights[col] = field.shape[0] - row
break
return int(np.sum(heights))