python 内存占用过多问题及其解决方案

1、问题背景
近期，一位 Python 开发者遇到了一个棘手的问题，他在开发过程中编写了一个能够穷举生成具有一定特征的矩阵的递归函数。然而，这个函数在运行时会占用过多的内存，导致服务器内存不足而被终止。

2、解决方案

为解决以上问题，该开发者尝试了以下方法：

（1）避免矩阵副本的内存引用。在 heavies() 函数中，每次生成的矩阵都会被复制一份副本，然后继续生成更多的矩阵。这种方式会导致大量的副本占据内存，从而导致内存占用过高。为了解决这个问题，可以在函数中使用一种叫做“生成器”（generator）的特殊函数类型。生成器可以生成一组值，但只在需要时才计算这些值。这样就可以避免生成大量的副本，从而减少内存占用。

import numpy as np

def heavies(row_sums, col_sums, col_index, mat_h):
    if col_index == len(col_sums) - 1:
        for stuff in heavy_col_permutations(row_sums, col_sums, col_index):
            mat_h[:, col_index] = stuff[0]
            yield mat_h.copy()
        return

    for stuff in heavy_col_permutations(row_sums, col_sums,