python - 使用 numpy 的任务

Question

给出了一个大小为 5x5 的整数矩阵。将此矩阵中第一行的所有负元素替换为数字 0。

我知道您需要循环遍历矩阵，但我不完全了解如何执行此操作。请解释一下，因为我想学习如何解决这些问题。

标签： pythonnumpy

Answer 1

 import numpy as np
 #create matrix of size (5,5) with all zeros
 matrix = np.zeros((5,5))
 matrix[0]=[1,-2,3,-4,-5]
 matrix[0]=np.where(matrix[0]<0,0,matrix[0])

说明：导入 numpy 后，初始化一个大小为 5x5 的 0 矩阵并分配一些值（+ve 和 -ve 到第一行）。现在将 np.where() 用于矩阵的第一行，将 0 分配给 -ve 值和 +ve 值保持不变。np.where() 将第一个参数作为条件，第二个参数是如果条件为真则做什么，否则当第三个参数中的条件为假时做什么