Python Pandas: Create dataframe based on condition

Question

I wanted to create a new dataframe on pandas based on conditions of the current dataframe. If the current dataframe is > 1000 rows then make the data frame empty (delete rows). Otherwise then just use the current dataframe. Below is my code that is having an error:

DF_NEW = np.where(len(DF)>1000, pd.DataFrame(DF[0:0]), pd.DataFrame(DF))

Corralien · Accepted Answer · 2022-07-28 07:04:54Z

1

You should use:

DF_NEW = pd.DataFrame(columns=DF.columns) if len(DF) > 1000 else DF.copy()

np.where need a vector with the same shape of your condition vector.

answered Jul 28, 2022 at 7:04

Corralien

121k8 gold badges43 silver badges69 bronze badges

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

Comments

bpfrd · Accepted Answer · 2022-07-28 07:05:14Z

1

Use if else as:

if len(DF)>1000:
    DF_NEW = pd.DataFrame(columns=DF.columns)
else:
    DF_new = DF.copy()

answered Jul 28, 2022 at 7:05

bpfrd

1,0255 silver badges11 bronze badges

Comments

user19580074 · Accepted Answer · 2022-07-28 07:12:00Z

0

ìf len(df) > 1000:
    for label in df.columns: 
        df.pop('label')
elif len(df) <= 1000:
    pass

your current code will make an array where, if the condition is True, it will take the first, and if condition is False, the latter

answered Jul 28, 2022 at 7:12

user19580074

Collectives™ on Stack Overflow

Python Pandas: Create dataframe based on condition

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related