Exercise

Generating permutation samples

As you worked out in the last exercise, we need to generate a permutation sample by randomly swapping corresponding entries in the semi_times and final_times array. Write a function with signature swap_random(a, b) that returns arrays where random indices have the entries in a and b swapped.

Instructions

100 XP
  • Define a function with signature swap_random(a, b) that does the following.
    • Create an array swap_inds the same length as the input arrays where each entry is True with 50/50 probability. Hint: Use np.random.random() with the size=len(a) keyword argument. Each entry in the result that is less than 0.5 should be True.
    • Make copies of a and b, called a_out and b_out, respectively using np.copy().
    • Use Boolean indexing with the swap_inds array to swap the appropriate entries of b into a_out and of a into b_out.
    • Return a_out and b_out.