遍历数据框中的列以添加到 Python pandas 中的 Total Value 列

Question

我正在尝试遍历某些列名以将“TotalValue”列中的总值相加。 我得到：

KeyError: 'TotalValue'

我已经展示了如果我输入它来得到总数的等式会是什么样子。

df['TotalValueByHand'] = df['Value_SCHD'] + df['Value_VBR'] + df['Value_IXUS'] + df['Value_MDIZX']

不过，我想遍历列，因为最终我计划在列表中添加更多内容。 抱歉，如果缩进很奇怪，我输入时看起来还不错。

stockList = ["SCHD", "VBR", "MDIZX", "IXUS"]

for s in stockList:
  
    df['TotalValue'] = np.where(df['Value_' + s] > 0, df['TotalValue'] + df['Value_' + s], df['TotalValue'])

Answer 1

如果只需要sum大于0的值，请使用列表理解生成所有列名称，并使用DataFrame.clip和sum ：

stockList = ["SCHD", "VBR", "MDIZX", "IXUS"]

cols = [f'Value_{s}' for s in stockList]
df['TotalValue'] = df[cols].clip(lower=0).sum(axis=1)

如果只需要sum ：

cols = [f'Value_{s}' for s in stockList]
df['TotalValue'] = df[cols].sum(axis=1)

如果创建由0填充的第一列，您的解决方案是可能的：

df['TotalValue'] = 0

for s in stockList:

    df['TotalValue'] = np.where(df['Value_' + s] > 0, df['TotalValue'] + df['Value_' + s], df['TotalValue'])