Как исправить NumPy ValueError для неоднородной формы массива? ⇐ Python

[Anonymous]

В этом разделе кода я пытаюсь случайным образом выбрать либо тету, либо мю равным нулю. Когда одна переменная равна нулю, мне нужна равномерно рандомизированная другая (и наоборот).

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

N = 10000

random = np.arccos(np.random.uniform(-1, 1, N))
zero = 0
choices = [random, zero]
theta = np.random.choice(choices)

if theta == random:
mu = zero
else:
mu = random

Я знаю, что случайные и ноль не имеют однородной формы. Вот почему я получил ошибку ValueError: установка элемента массива с последовательностью. Запрошенный массив имеет неоднородную форму после 1 измерения. Обнаруженная форма была (2,) + неоднородная часть. Однако я не знаю, как это исправить (я еще новичок в программировании). Любые мысли будут оценены.

Подробнее здесь: https://stackoverflow.com/questions/782 ... rray-shape