Как преобразовать массив в новый со следующими значениями?

Question

[Vechnyy]

Как преобразовать массив в новый через цикл for со следующими значениями?

* "small", если значения меньше 20
* "medium", если значения в промежутке [20, 30]
* "large", если значения больше 30

# создадим трехмерный массив для примера
a_random = np.random.randint(15, 37, (2, 3, 4))
print(a_random)

Answer 1

[o5a]

b_random = np.select([a_random < 20, a_random <= 30, a_random > 30], ["small", "medium", "large"])

Comments

[Vechnyy]

Это не ответ т.к. решение должно было происходить через for

[o5a]

Vechnyy, тогда можно так

b_random = np.empty_like(a_random, dtype=object)

for a, b in np.nditer([a_random, b_random], op_flags=['readwrite'], flags=['refs_ok']):
    b[...] = 'small' if a < 20 else 'medium' if a <= 30 else 'large'

print(b_random)

Answer 2

[Wataru]

print ([("small" if a < 20 else "medium") if a <=30 else "large" for a in a_random])

Comments

[Vechnyy]

Ошибка:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

[Wataru]

Vechnyy, Это потому что у вас массив двумерный, а я для одномерного привел. Подумайте, как можно пройтись по двумерному массиву? Или, может, превратить его в одномерный и потом назад?

[Vechnyy]

Я вроде трехмерный создавал. Там 3 вложения или что-то перепутал?

[Wataru]

Vechnyy, Трехмерный, да. Я запутался. Посмотрите в сторону Numpy.reshape().

Ну, или вместо list comprehension делайте 3 цикла и append. Или 3 цикла по координатам и заменяйте каждый элемент отдельно, но это не очень по-питонски.

[Vechnyy]

Илья Николаевский, это я для практики )

Answer 3

[Владимир Куц]

Как преобразовать массив в новый через цикл for со следующими значениями

В отмеченном ответе цикла for не увидел.
Вообще такое решается рекурсией:

def mark(data):
    if isinstance(data, (int, np.int32)):
        if data < 20:return 'small'
        elif data <= 30:return 'medium'
        else:return 'large'
    elif isinstance(data, (list, np.ndarray)):
        return [mark(item) for item in data]   #  а вот и цикл for

a_random = np.random.randint(15, 37, (2, 3, 4))
print(a_random)
    
print(mark(a_random))

# [[[28 30 35 35]
#  [16 31 35 34]
#  [22 16 29 27]]
#
# [[23 35 24 19]
#  [26 21 22 28]
#  [30 25 36 15]]]
# [[['medium', 'medium', 'large', 'large'], ['small', 'large', 'large', 'large'], ['medium', 'small', 'medium', 'medium']], [['medium', 'large', 'medium', 'small'], ['medium', 'medium', 'medium', 'medium'], ['medium', 'medium', 'large', 'small']]]

Comments

[Vechnyy]

Не подскажите почему у меня "None" пишет везде в массиве?

[Владимир Куц]

Vechnyy, вероятно у вас другой тип используется.
Перечислите тип элементов, из которых ваш массив состоит в этой строке:
if isinstance(data, (int, np.int32)):
и все заработает