Как преобразовать элементы в массиве 0-d numpy?

Question

Итак, скажем, у меня есть массив 0-d, подобный этому:

a = [3.0, 0.0, 3.0, 0.0, 5.0]

для каждого элемента в этом массиве, я хочу преобразовать его обратно в определенное строковое значение в зависимости от числа. Поэтому для 0,0 я хочу вернуть «нет». Для 3.0 я хочу вернуть «может быть». А для 5.0 хочу ответить «да». Это будет выглядеть так:

a = ['maybe', 'no', 'maybe', 'no', 'yes']

моя проблема в том, что когда я пытаюсь написать для l oop для преобразования каждого элемента, я не думаю, что могу использовать enumerate, поскольку все элементы хранится в массиве 0-d. Я не знаю, как еще go я могу обновить этот тип массива.

Answer 1

Вы ищете это:

a = np.array([3.0, 0.0, 3.0, 0.0, 5.0])
dictionary = {0:'no',3:'maybe',5:'yes'}
[dictionary[i] for i in a]

вывод:

['maybe', 'no', 'maybe', 'no', 'yes']