assign random values to all rows based on another column

Question

I want to assign random numerical values ranging from 600-1200 to my column df['size'] but based on another column. So if the value of the same row in df['type']is yes, the value of df['size'] should be in a range from 600-1200 but if the value of df['type']is "no", then the value of df['size'] should be a random value between 1500-2000.

df['size'] = np.random.randint(600,1200, size=len(df))

jezrael · Accepted Answer · 2021-03-15 14:48:47Z

2

Use numpy.where for select if matched yes or not:

a = np.random.randint(600,1200, size=len(df))
b = np.random.randint(1500,2000, size=len(df))
df['size'] = np.where(df['type'].eq('yes'), a, b)

answered Mar 15, 2021 at 14:48

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

assign random values to all rows based on another column

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related