首页python在矩阵首行添加一行0数据

python在矩阵首行添加一行0数据

时间: 2024-01-21 07:16:27 浏览: 84

python 实现矩阵置零

# 给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。 # 示例 1: # 输入: # [ # [1,1,1], # [1,0,1], # [1,1,1] # ] # 输出: # [ # [1,0,1], # [0,0,0], # [1,0,1] # ] # 示例 2: # 输入: # [ # [0,1,2,0], # [3,4,5,2], # [1,3,1,5] # ] # 输出: # [ # [0,0,0,0], # [0,4,5,0], # [0,3,1,0] # ]

可以使用numpy库中的insert函数来在矩阵的首行添加一行0数据。具体代码如下： ```python import numpy as np # 创建一个3x3的矩阵 matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) # 在首行添加一行0数据 new_row = np.array([0, 0, 0]) matrix = np.insert(matrix, 0, new_row, axis=0) # 打印新矩阵 print(matrix) ``` 输出结果为： ``` array([[0, 0, 0], [1, 2, 3], [4, 5, 6], [7, 8, 9]]) ```

阅读全文