Код Middle Хи-квадрат и Манна-Уитни обновлено
Статистика U Манна-Уитни на Python
Условие
Напишите mann_whitney_u(a, b): статистика U для группы a (число пар, где элемент a больше элемента b, ничья даёт 0.5) и оценка P(A > B) как U / (len(a) * len(b)). Верните [U, round(prob, 4)], U как float.
Примеры
mann_whitney_u([3, 5, 7], [1, 2, 6]) // [7, 0.7778]
mann_whitney_u([1, 2], [3, 4]) // [0, 0]
mann_whitney_u([5, 5], [5, 5, 5]) // [3, 0.5]
Как подойти
Подсказки: 2
- Перебор всех пар
(x, y). - Ничья добавляет 0.5.
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.
Предыдущий вопрос
Статистика хи-квадрат для таблицы сопряжённости
Следующий вопрос
Проверка SRM: хи-квадрат и p-value