Как исправить код?

Question

[minamisa]

import numpy as np
arr = np.arange(10).reshape(10)
print(arr)
arr_max = np.max(arr)
arr_mean = np.mean(arr)
arr1 = np.where(arr_max,arr_mean, arr)
print(arr1)
print(arr_max)
print(arr_mean)

[11.1 11.1 11.1 11.1 11.1 11.1 11.1 11.1 11.1 11.1]
Выводит вот это, а мне надо, чтобы максимальное значение заменялось на среднее. Помогите, пожалуйста, сделать правильно

Answer 1

[artem]

import numpy as np

arr = np.arange(10).reshape(10)
print(arr)

arr_max = np.max(arr)
arr_mean = np.mean(arr)

arr1 = np.where(arr == arr_max, arr_mean, arr)
print(arr1)

print(arr_max)
print(arr_mean)

в строке с np.where нужно сравнивать arrс arr_max, чтобы заменить максимальное значение на среднее