python - 从 for 循环到列表理解

Question

我有一个显示和 list 报告的程序，我只是想知道如何将以下内容放入列表理解而不是 for 循环中......我对所有这些 python 行话有点菜鸟但是来 self 所知道的是，任何形式为 for 循环的东西也可以表示为列表理解....任何帮助将不胜感激

def rowSum(TotSize,data,row,col):
    """Calculates the sum of each row in a given 2 dimensional list and stores
it into a given one dimensional list"""
    for i in range(row):
        sum = 0
        for j in range(col):
            sum += data[i][j]
        TotSize[i] = sum

Answer 1

你的代码本质上等同于

TotSize[:] = map(sum, data)

这将对所有 data 求和，而不仅仅是第一 row 行和第一 col 列。它还会调整 TotSize 的大小以匹配 data 的行数(假设 TotSize 是一个列表)。

我想知道你为什么要传递应该存储结果的列表。在 Python 中，您通常会简单地返回该列表:

def row_sums(data):
    return map(sum, data)

现在值得怀疑是否值得为此定义一个函数......