using pandas Creating a new column after every For loop iteration in Python

Question

Input is like below csv file

    y   proba
1   1    12
2   1    65
3   1    1
4   1    54


for i in range(len(df1["proba"])):
  a=pd.concat([df1, pd.Series(df1["proba"] >= df1["proba"][i], name=f'tau{i}').astype(int)], axis=1)
  print(a)

I apply this but wont get answer correctly

Need a answer like below

   y  proba   tau1    tau2    tau3    tau4    
1   1   12      1       0       1       0       
2   1   65      1       1       1       1       
3   1   1       0       0       1       0       
4   1   54      1       0       1       1

Andrej Kesely · Accepted Answer · 2022-09-17 14:23:38Z

1

Try:

for i, v in enumerate(df["proba"]):
    df[f"tau{i+1}"] = (df["proba"] >= v).astype(int)

print(df)

Prints:

   y  proba  tau1  tau2  tau3  tau4
1  1     12     1     0     1     0
2  1     65     1     1     1     1
3  1      1     0     0     1     0
4  1     54     1     0     1     1

answered Sep 17, 2022 at 14:23

Andrej Kesely

196k15 gold badges60 silver badges105 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Control Solution · Accepted Answer · 2022-09-17 14:54:01Z

0

new = df1.copy()
for i in range(len(df1["proba"])):
  a=pd.DataFrame(pd.Series(df1["proba"] >= df1["proba"][i], name=f'tau{i}').astype(int))
  new.insert(len(new.columns), f'tau{i}', a.values)

answered Sep 17, 2022 at 14:54

Control Solution

313 bronze badges

Collectives™ on Stack Overflow

using pandas Creating a new column after every For loop iteration in Python

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related