import pandas as pd
# Set up information on the number of people requested for each store
dict_req = {
"location": ["tenjin", "hakata", "akasaka", "gakken"], # Store name
"num_managers": [
1,
1,
1,
1,
], # Number of required employees with manager position at each store
"num_submanagers": [
1,
0,
1,
1,
], # Number of requested employees with assistant manager position at each store
"num_employees_any_position": [
2,
2,
2,
2,
], # Number of employees requested for each store (all employees with or without managerial position)
}
df_req = pd.DataFrame.from_dict(dict_req, orient="index").T
# Set each employee's work location preference
dict_worker_loc = {
# ID of the employee
"worker_id": [0, 1, 2, 3, 4, 5, 6, 7, 8],
# Employee's work location preference for tenjin store
"tenjin": [2, 0, 0, 0, 1, 1, 2, 1, 1],
# Employee's work location preference for hakata store
"hakata": [1, 0, 0, 2, 2, 2, 1, 2, 1],
# Employee's work location preference for akasaka store
"akasaka": [1, 0, 0, 1, 0, 1, 1, 1, 2],
# Employee's work location preference for gakken store
"gakken": [1, 2, 2, 0, 0, 0, 0, 0, 0],
}
df_worker_loc = pd.DataFrame.from_dict(dict_worker_loc, orient="index").T
# Set each employee's position qualifications
dict_worker_skill = {
# ID of the employee
"worker_id": [0, 1, 2, 3, 4, 5, 6, 7, 8],
# Qualified as a manager 1 or not 0
"manager": [1, 1, 0, 0, 1, 1, 1, 0, 1],
# Qualified as an assistant manager 1 or not 0
"submanager": [1, 1, 1, 0, 1, 1, 1, 0, 1],
# Qualified as non-managerial position 1 or not 0
"staff": [1, 1, 1, 1, 1, 1, 1, 1, 1],
}
df_worker_skill = pd.DataFrame.from_dict(dict_worker_skill, orient="index").T