Найти среднее количество ископаемых с numpy

import random
import time
 
class Terrain():
   
    description = {
        0 : "М'яка гірська порода", 
        1 : "Тверда гірська порода",
        2 : "Надтверда гірська порода",
        3 : "Вода",
        4 : "Нафта",
        5 : "Кам'яне вугілля",
        6 : "Залізна руда",
        7 : "Золото",
    }
    def __init__(self, m, n, k):
        self.array = [[[random.randint(0, 7) for i in range(m)] for j in range(n)] for l in range(k)]
        
    
    def get_all_reses(self):
        layers = {}
        count = 0
        reses = {0:0, 1:0, 2:0, 3:0, 4:0, 5:0, 6:0, 7:0}
        for layer in self.array:
            for row in layer:
                for i in row:
                    reses[i] += 1
        return reses
    
    def get_layers_number(self):
        return len(self.array)
    
    
    def get_cols_number(self,):
        
        if len(self.array) > 0:
            return len(self.array[0]) * len(self.array[0][0])
            
        else:   
            return 0
        
        
    def get_average_in_layers(self):
        reses = self.get_all_reses()
        coef = self.get_layers_number()
        print("Середня кількість кожної копалини в пласті грунту: \n")
        for i in reses:
            print(self.description[i]+": ", reses[i]/coef)
        
        
    def get_average_in_columns(self):
        reses = self.get_all_reses()
        coef = self.get_cols_number()
        print("Середня кількість кожної копалини в одному стовпці грунту: ")
        for i in reses:
            print(self.description[i]+": ", reses[i]/coef)
            
    def get_resources_in_terrain(self):
        reses = self.get_all_reses()
        coef = self.get_cols_number()
        print("Кількість копалин у всьому грунті: ")
        for i in reses:
            print(self.description[i]+": ", reses[i])
            
    def __str__(self):
        string = ""
        for i in self.array:
            string += str(i) + "\n"
        return string
        
        
        
 
print("Початок роботи програми \n")
generated = False
while True:
    if not generated:
        print("Усі значення повинні бути більше за 0.")
        m = int(input("Введіть ширину массиву: "))
        n = int(input("Введіть довжину массиву: "))
        k = int(input("Введіть глибину массиву: "))
        if m != 0 and n != 0 and k != 0:
            generated_array = Terrain(m, n, k)
            generated = True
 
    ask = int(input("""\n 1. Отримати середню клількість копалин на кожному шарові.
                   \n 2. Отримати середню кількість копалин на кожному стовпці.
                   \n 3. Отримати загальну кількість копалин у всьому грунті.
                   \n 4. Переглянути весь массив.
                   \n 5. Перегенерувати массив.
                   \n Оберіть варіант: """))
    if ask == 1:
        generated_array.get_average_in_layers()
    elif ask == 2:
        generated_array.get_average_in_columns()
    elif ask == 3:
        generated_array.get_resources_in_terrain()
    elif ask == 4:
        print(generated_array)
    elif ask == 5:
        generated = False