Life's a garden. Dig it.
You track the whereabouts of 100 individuals in a DataFrame called whereabouts. Each person has a corresponding list of place ids indicating the places they’ve visited in the recent week. You also track which places have been exposed to COVID-19 in a list called exposed 😷.
import numpy as np
import pandas as pd
# exposed places
exposed = [0,5,9]
# whereabouts of each person
generator = np.random.default_rng(2468)
Nplaces = 10
Npersons = 10
place_ids = np.arange(Nplaces)
visits = generator.choice(place_ids, size=3*Nplaces, replace=True)
split_idxs = np.sort(generator.choice(len(visits), size=9, replace=True))
whereabouts = pd.DataFrame({
'person_id': range(Npersons),
'places': [np.unique(x).tolist() for x in np.array_split(visits, split_idxs)]
})
print(whereabouts)
# person_id places
# 0 0 [3, 4, 5, 6]
# 1 1 []
# 2 2 [3]
# 3 3 [6, 8, 9]
# 4 4 [3]
# 5 5 [0, 2, 5, 6, 7, 8]
# 6 6 [2, 7]
# 7 7 [0, 5, 8, 9]
# 8 8 [2, 7, 9]
# 9 9 [0, 5, 8, 9]For each person, identify the places they visited which have been exposed. Make this a new list-column in whereabouts called exposures.
Solution 1¶
This content is gated
Subscribe to one of the products below to gain access
Solution 2¶
This content is gated
Subscribe to one of the products below to gain access