python - 遍历 Pandas 数据框中的多个列并创建新列

Question

我有一个包含度量列 SalesMonth1 -- SalesMonth12 和另一列 Index 和 price 的数据框。

我正在尝试执行以下操作，但它不起作用。你能推荐一个更好的方法吗？

For i in range(12):
     DF['Newcol'] = np.where(DF["Index"] >0,
                             DF["SalesMonth[i]"] * DF["price"],
                             DF["SalesMonth[i]"])

Answer 1

你很接近。试试这个:

for i in range(12):
    df['newcol'+str(i)] = np.where(df['Index'] > 0,
                                   df['SalesMonth'+str(i)] * df['price'],
                                   df['SalesMonth'+str(i)])

诀窍是通过 str 将整数转换为字符串。