Python - 从矩阵中打印在给定索引处具有相同元素的行