Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement