Как проверить элементы массива на какое-то условие?

Question

[albus-feles]

Имеется случайный массив:

а = np.random.randint(15, 37, (2, 3, 4))

Используя этот массив, необходимо преобразовать его в новый массив со следующими значениями:

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

К примеру у нас есть массив:

[[[33 22 16 29]
[28 36 36 23]
[15 26 27 24]]

Нужно провести элементы под эти условия. На выходе получаем приблизительно такой массив:

[[[large medium small medium]
[medium large large medium]
[small medium medium medium]]

Не пойму, как числовой элемент провести под условие и получить вместо него значение типа str.

Answer 1

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

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

[[[33 22 16 29]
  [28 36 36 23]
  [15 26 27 24]]

 [[17 26 36 31]
  [28 20 33 32]
  [22 30 19 31]]]


b = np.full_like(a.astype(str), 'medium')

b[a < 20] = 'small'
b[a > 30] = 'large'

print(b)
[[['large' 'medium' 'small' 'medium']
  ['medium' 'large' 'large' 'medium']
  ['small' 'medium' 'medium' 'medium']]

 [['small' 'medium' 'large' 'large']
  ['medium' 'medium' 'large' 'large']
  ['medium' 'medium' 'small' 'large']]]

Comments

[albus-feles]

К сожалению, требуется другое решение:

К примеру у нас есть массив

[[[33 22 16 29]
[28 36 36 23]
[15 26 27 24]]

Нужно провести элементы под условие и получить приблизительно такой массив:
[[[large medium small medium]
[medium large large medium]
[small medium medium medium]]

Т.е, заменить элементы массива на определенную условием строку
Не пойму, как числовой элемент провести под условие и получить вместо него значение типа str

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

albus-feles, ну сделайте так:

b = np.full_like(a.astype(str), 'medium')

b[a < 20] = 'small'
b[a > 30] = 'large'

print(b)

[[['large' 'medium' 'small' 'medium']
['medium' 'large' 'large' 'medium']
['small' 'medium' 'medium' 'medium']]

[['small' 'medium' 'large' 'large']
['medium' 'medium' 'large' 'large']
['medium' 'medium' 'small' 'large']]]

[albus-feles]

Спасибо большое: буду разбираться с этим методом :)

[albus-feles]

Владимир, а можно ли сделать это задание без методов np.full_like и a.astype(str)?
К примеру через операторы if-elif else?

<a = np.random.randint(15, 37, (1, 3, 3))
for i in range(0,3):
  for j in range(0,3):
    print (a[0,i,j])
    if a[0,i,j] < 20:
       a[0,i,j] = 'small'
    elif 20< a[0,i,j] < 30:
      a[0,i,j] = 'medium'
    else:
      a[0,i,j] = 'large'
print(a)

Он конечно начинает ругаться.
Подскажите, пожалуйста, как в такой конструкции присвоить числовому значению матрицы буквенное?

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

Ну смотрите... Вы же используете numpy для генерации массива.
Массивы numpy отличаются от массивов python. Они типизированные, за счет чего скорость работы с ними выше. Соответственно брать массив целых чисел и вставлять туда строки - нельзя. Нужно создать отдельный типизированный массив строк.
Как вы можно сделать - не используя numpy, используя традиционные массивы python