Python|常用行元素求和

在Python中，我们经常需要对一维列表、二维列表等各种形式的行元素求和。本文将为大家介绍Python中求和的几种常见方式。

普通for循环求和

最基本的方式就是通过for循环，一个一个遍历列表中的元素，将其相加。这种方法简单易懂，但对于长度较长的列表效率较低。

lst = [1, 2, 3, 4, 5]
total = 0
for num in lst:
    total += num
print(total)

结果：

15

使用内置函数sum求和

Python中提供了内置函数sum，可以直接对一维列表中的所有元素求和，这种方法比for循环效率更高，是Python中常用的一种方式。

lst = [1, 2, 3, 4, 5]
total = sum(lst)
print(total)

结果：

15

使用numpy库进行求和

当我们需要对多维数组的行/列元素求和时，可以使用numpy库中的sum函数。numpy的sum函数功能更强大，不仅可以对行/列求和，还可以对整个矩阵中的所有元素求和。

import numpy as np

lst = [[1, 2], [3, 4], [5, 6]]  # 二维列表
arr = np.array(lst)  # 将列表转换成numpy数组

# 对行元素求和
row_sum = np.sum(arr, axis=1)
print(row_sum)

# 对列元素求和
col_sum = np.sum(arr, axis=0)
print(col_sum)

# 对整个数组中的所有元素求和
total_sum = np.sum(arr)
print(total_sum)

结果：

[ 3  7 11]
[ 9 12]
21

总结

以上是Python中常用的几种行元素求和的方法，其中sum函数是最常用的一种。如果需要对多维数组的行/列求和，可以使用numpy库中的sum函数，它比sum函数功能更强大。

需要注意的是，在使用numpy库进行求和时，需要将列表或数组转换成numpy数组，才能使用numpy中的函数。