Выборка из совместной функции массы вероятности в PythonPython

Программы на Python
Ответить
Anonymous
 Выборка из совместной функции массы вероятности в Python

Сообщение Anonymous »

У меня есть неотрицательный нормализованный вектор p. Я хотел бы выбрать индекс из набора индексов вектора. Вероятность получения выборки k равна p[k]. Используя функцию np.random.choice, я могу написать следующий код.

Код: Выделить всё

p = [0.2, 0.3, 0.1, 0.3, 0.1]
indices = np.arange(len(p))
k = np.random.choice(indices, p=p)
Мой вопрос: как я могу обобщить этот код для многомерных массивов? Например, учитывая трехмерный неотрицательный нормализованный тензор IxJxK p = np.random.rand(I,J,K)
как я могу выбрать индекс (i,j,k) с вероятностью p[i,j,k]?


Подробнее здесь: https://stackoverflow.com/questions/791 ... -in-python
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»