Как найти соотношение нескольких столбцов в пандах?

Question

У меня есть такие столбцы, как

total_balance, b1_amt, b2_amt, b3_amt, b4_amt, b5_amt 

, и всего их более 100 - как найти соотношение каждого столбца в новых столбцах?

Например, я попытался

df['ratio_b1]=df[b1_amt]/total_balance 

, но если я применяю этот подход, то я должен написать код для каждого столбца, поскольку существует более 100 столбцов, так как я могу написать все это водин код?

Answer 1

Если ваши столбцы все названы последовательно с использованием точного шаблона в вашем вопросе, вы можете сделать что-то вроде этого:

for i in range(1, 101):
    df['ratio_b{}'.format(i)] = df['b{}_amt'.format(i)] / df['total_balance']