程序消耗的 memory 比 Python 中声明的要多

Question

问题：给定一个二维矩阵，我们需要找出该矩阵中给定范围（包括）之间有多少行和列的总和。 所有这些范围都将以一些查询的形式给出。

样本输入 -> 矩阵 - [[2,3],[5,9]]，范围查询 - [[1,7],[6,14]] Output - [2,3]

解释 -> 对于第一个范围查询，即 [1,7]，我们有第一行（2+3=5）和第一列（2+5=7），范围之间的总和。 所以 output 将为 2。对于第二个查询，即 [6,14]，我们有第一列（2+5=7），第二列（3+9 = 12），第二行（5+9=14），所以 output将是 3。

约束：1<=nxm<=4x10^6（矩阵大小），1<=q<=5x10^5（查询数），1<=Aij<=10^9（矩阵元素），1< =L<=R<=10^14（查询范围）

代码：

def solve (a, queries):
    # Write your code here
    #print('y')
    rows = np.sum(a, axis=0)
    cols = np.sum(a, axis=1)
    
    for query in queries:
        low = query[0]
        high = query[1]

        r = np.sum((low<=rows) * (rows<=high)) 
        c = np.sum((low<=cols) * (cols<=high)) 

        yield r+c
    

N = int(input())
M = int(input())
a = [list(map(int, input().split())) for i in range(N)]
Q = int(input())
S = int(input())
query = [list(map(int, input().split())) for i in range(Q)]

out_ = solve(a, query)
print (' '.join(map(str, out_)))```


*For the above sample input it is consuming 56304 KiB memory and I haven't declared
such a big memory in my program.
For larger cases memory limit exceeds 256 MB.
I don't know what is causing this problem.
Please help.*