How can I drop column using combination of string and integer?

Question

I have many columns name with string with (t+1) (t+2) .... (t+54) In pictured attached, How can I drop just only columns "lemon(t+1),....,lemon(t+30), not lemon(t-1)

Thanks

enter image description here

shmakovpn · Accepted Answer · 2020-05-26 12:35:47Z

1

See comments in code

cols = df.columns  # get columns of the dataframe
# filter the list of columns by removeing items from 'lemon(t+1)' to 'lemon(t+30)'
cols2 = filter(lambda x: not (x.startswith('lemon(t+') and x[8:-1]>0 and x[8:-1]<31) )
df_result = df[cols2]  # create needed dataframe

answered May 26, 2020 at 12:35

shmakovpn

85110 silver badges12 bronze badges

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

Comments

Umar.H · Accepted Answer · 2020-05-26 12:49:32Z

1

using a list comprehension -

df.drop([f"lemon(t-{i})" for i in range(1,31)],axis=1)

edited May 26, 2020 at 12:49

answered May 26, 2020 at 12:19

Umar.H

23.1k7 gold badges50 silver badges94 bronze badges

Collectives™ on Stack Overflow

How can I drop column using combination of string and integer?

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