Skip to content

Instantly share code, notes, and snippets.

@gonzalezgouveia
Created March 18, 2019 20:31
Show Gist options
  • Save gonzalezgouveia/075d8671491e6e5170fb585389c4841e to your computer and use it in GitHub Desktop.
Save gonzalezgouveia/075d8671491e6e5170fb585389c4841e to your computer and use it in GitHub Desktop.
def select_name(gender):
'''select a name based on gender'''
if gender == 'M':
name = np.random.choice(name_men_list)
elif gender == 'F':
name = np.random.choice(name_women_list)
return(name.split(' ')[0].capitalize())
def create_table(n):
'''create a pandas dataframe given a desired number of rows (n)'''
age = np.random.randint(low_age, high_age, size=n)
height = np.around(np.random.uniform(low_height, high_height, size=n),
decimals)
gender = np.random.choice(genders, size=n)
name = [select_name(x) for x in gender]
df = pd.DataFrame({
'age':age,
'height':height,
'gender':gender,
'name':name
})
return(df)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment