请求解决python中零矩阵的一个问题

将一个N*M矩阵中 元素0所在的行和列都设置为0

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        n=len(matrix)
        m=len(matrix[0])
        i=j=k=0
        store=[]
        while i<n:
            while j<n:
                if matrix[i][j]==0:
                    store.append([i,j])               #存储0 所在的确切位置
                    j=j+1
                else:
                    j=j+1
            j=0
            i=i+1

        i=0
        n_1=n


        while i<len(store):
            a_1=store[i][0]
            a_2=store[i][1]
            for num_1 in range(m):                                                  #将一行变为全是0
                matrix[a_1][num_1]=0
            for num_2 in range(n):                                                    #将一列变为全是0
                matrix[num_2][a_2]=0
            
            i=i+1

始终解决不了列的问题，一直没想出来问题在哪

img