python行列求和
时间: 2023-10-29 17:57:41 浏览: 31
在Python中,可以使用不同的方法来对行和列进行求和。
对于行向求和,可以定义一个函数`row_sum`,该函数接受一个二维数组作为输入,表示图像,然后对每一行进行求和,并将结果存储在一个列表中。具体实现如下:
```python
def row_sum(img):
h_m, w_m = img.shape #[1]
r_sum = []
for r in range(h_m):
sum_crt_r = sum(img[r])
r_sum.append(sum_crt_r)
return r_sum
```
这个函数首先获取图像的高度和宽度,然后遍历每一行,使用`sum`函数对该行进行求和,将结果存储在列表`r_sum`中,并最终返回该列表。
对于列向求和,可以定义一个函数`col_sum`,该函数接受一个二维数组作为输入,表示图像,然后对每一列进行求和,并将结果存储在一个列表中。具体实现如下:
```python
def col_sum(img):
h_m, w_m = img.shape #[2]
c_sum = np.zeros(w_m, dtype=np.int32)
for c in range(w_m):
c_sum += img[:, c]
col_sums = c_sum.tolist()
return col_sums
```
这个函数首先获取图像的高度和宽度,并创建一个与图像宽度相同的全零数组`c_sum`,然后遍历每一列,将该列的元素累加到`c_sum`数组中,最后将`c_sum`数组转换成列表`col_sums`并返回。
这两个方法可以分别用于行和列的求和,并且可以应用到其他类似的问题上。