📜  Python|嵌套列表的逐列总和(1)

📅  最后修改于: 2023-12-03 15:34:19.652000             🧑  作者: Mango

Python | 嵌套列表的逐列总和

在Python中,可以使用嵌套列表来表示矩阵或数组,有时需要对每一列进行求和操作。本文介绍了如何使用Python对嵌套列表的每一列进行求和。

方法一:使用for循环

可以使用for循环遍历嵌套列表,对每一列进行求和操作。

# 定义嵌套列表
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]

# 初始化列表
sums = [0, 0, 0]

# 遍历列表
for row in matrix:
    for i in range(len(row)):
        sums[i] += row[i]

print(sums) # 输出[12, 15, 18]
方法二:使用numpy库

numpy是一个Python的科学计算库,可以使用numpy.sum()对嵌套列表进行每一列求和操作。

# 导入numpy库
import numpy as np

# 定义嵌套列表
matrix = np.array([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
])

# 对每一列求和
sums = matrix.sum(axis=0)

print(sums) # 输出[12, 15, 18]
总结

本文介绍了两种方法对Python中的嵌套列表进行每一列求和操作。使用for循环的方法适用于简单的嵌套列表,numpy库适用于矩阵或数组的计算。在实际开发中,应根据实际情况选择合适的方法。