Generate values in R and Python

Question

I am trying to generate dummy data with some probability. Let say I want to have dummy data about people by gender. I already prepare this in R and you can see my code line below.

gender = sample(x=c("M","F"), prob = c(.6, .4),size=100,replace=TRUE)

Now I want to prepare the same thing but now in Python in Pandas Data Frame. Can anybody help me how to solve this problem?

score 2 · Accepted Answer · answered Sep 19 '22 at 14:47

You can use numpy.random.choice , replace is True by default.

>>> np.random.choice(a=["M", "F"], size=100, p=[0.6, 0.4])

array(['F', 'M', 'F', 'M', 'M', 'M', 'M', 'M', 'M', 'F', 'M', 'F', 'M',
       'M', 'F', 'M', 'M', 'M', 'F', 'F', 'M', 'M', 'F', 'F', 'M', 'F',
       'F', 'M', 'M', 'F', 'F', 'M', 'F', 'F', 'M', 'F', 'M', 'M', 'F',
       'M', 'M', 'F', 'F', 'M', 'M', 'F', 'M', 'M', 'F', 'M', 'M', 'M',
       'F', 'F', 'M', 'F', 'M', 'M', 'M', 'M', 'M', 'M', 'F', 'F', 'M',
       'M', 'F', 'F', 'M', 'F', 'M', 'F', 'M', 'M', 'F', 'M', 'F', 'F',
       'M', 'F', 'M', 'F', 'M', 'F', 'M', 'M', 'F', 'M', 'M', 'M', 'F',
       'F', 'F', 'F', 'F', 'M', 'M', 'F', 'F', 'F'], dtype='<U1')

The Thonnu · Answer 2 · 2022-09-19T15:05:56.437

1

Try this. random.choices gets k choices from the iterable provided:

import random
print(random.choices("MF", weights=[.6,.4], k=100))

Testing:

>>> l = random.choices("MF", weights=[.6,.4], k=100)
>>> l
['M', 'F', 'F', 'M', 'M', 'M', 'M', 'M', 'F', 'M', 'M', 'F', 'M', 'M', 'M', 'F', 'M', 'M', 'M', 'M', 'F', 'M', 'M', 'M', 'F', 'F', 'M', 'F', 'F', 'M', 'M', 'F', 'M', 'M', 'M', 'M', 'M', 'M', 'M', 'M', 'M', 'M', 'F', 'M', 'M', 'F', 'M', 'F', 'M', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'M', 'F', 'F', 'M', 'M', 'M', 'F', 'F', 'M', 'M', 'M', 'F', 'F', 'F', 'M', 'F', 'F', 'M', 'M', 'F', 'F', 'M', 'M', 'M', 'F', 'M', 'M', 'F', 'M', 'M', 'M', 'M', 'M', 'F', 'M', 'M', 'M', 'F', 'F', 'F', 'M', 'F', 'F', 'M']
>>> l.count("M")
60
>>> l.count("F")
40

edited Sep 19 '22 at 15:05

answered Sep 19 '22 at 14:49

The Thonnu

3,578
2
8
30

It doesn't take probabilities in account – ThePyGuy Sep 19 '22 at 14:51
I don't see it being fixed – ThePyGuy Sep 19 '22 at 14:53
@ThePyGuy - oops, I updated the code, but not the output. I've updated both now. – The Thonnu Sep 19 '22 at 14:54
Can you tell me how it takes care of probabilities `.6, .4`? – ThePyGuy Sep 19 '22 at 15:02
There is `weights` parameter in `random.choice` which you can probably use to define the probabilities, docs at https://docs.python.org/3/library/random.html#random.choice – ThePyGuy Sep 19 '22 at 15:04
@ThePyGuy - it's because there are 3 `M`s and 2 `F`s, so `M` has a 60% chance of appearing, and `F` has a 40% chance – The Thonnu Sep 19 '22 at 15:06
But, the weights parameter is probably better so I updated my answer. – The Thonnu Sep 19 '22 at 15:07

Generate values in R and Python

2 Answers2