在数据框中，如何根据另一列的条件在另一列中写入字符串

Question

我在 excel 中有一个会计试算表 (tb)，我在 python 中加载为 df。 df 在左侧有一个索引列，从 0 开始。 第 1 列是帐户名称，第 2 列是借方，第 3 列是贷方。 每行是一个具有借方或贷方余额的帐户。 tb 的末尾有一个“总计”行，第 2 列和第 3 列中有借方和贷方的总计。借方和贷方的总计彼此相等。

我想创建一个新的“帐户编号”列并为每个帐户分配一个编号。这些数字应该是从 1 到有多少帐户，最多不包括总计行。总计行中的帐户 # 应为空白。

我编写了以下 for 循环，但它继续在总计行中分配一个帐号：

for i in df['Account']:
    if i == 'TOTAL':
        df['account #'] == ''
    else:
        df['account #'] = df.index + 1

我该怎么做才能让“totals”的帐户 # 为空白？

Answer 1

我们可以的

import numpy as np 

df['account #']=np.where(df['Account']=='TOTAL', '' , df.index+1)