НачатьНачать бесплатно

Генерация перестановочной выборки

В видеоуроке вы узнали, что перестановочное семплирование — отличный способ смоделировать гипотезу о том, что две переменные имеют одинаковое распределение вероятностей. Это распространённая гипотеза для проверки, поэтому в данном упражнении вы напишете функцию для генерации перестановочной выборки из двух наборов данных.

Напомним: перестановочная выборка из двух массивов, содержащих n1 и n2 элементов соответственно, строится следующим образом — массивы объединяются в один, содержимое объединённого массива перемешивается, после чего первые n1 элементов образуют перестановочную выборку первого массива, а последние n2 элементов — второго.

Это упражнение является частью курса

Статистическое мышление на Python (часть 2)

Посмотреть курс

Инструкции к упражнению

  • Объедините два входных массива в один с помощью np.concatenate(). Обязательно передайте data1 и data2 как один аргумент (data1, data2).
  • Используйте np.random.permutation(), чтобы перемешать объединённый массив.
  • Сохраните первые len(data1) элементов массива permuted_data в переменную perm_sample_1, а последние len(data2) элементов — в perm_sample_2. Для этого используйте срезы :len(data1) и len(data1): применительно к permuted_data.
  • Верните perm_sample_1 и perm_sample_2.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

def permutation_sample(data1, data2):
    """Generate a permutation sample from two data sets."""

    # Concatenate the data sets: data
    data = ____

    # Permute the concatenated array: permuted_data
    permuted_data = ____

    # Split the permuted array into two: perm_sample_1, perm_sample_2
    perm_sample_1 = permuted_data[____]
    perm_sample_2 = permuted_data[____]

    return perm_sample_1, perm_sample_2
Редактировать и запускать код