У меня есть 2D-массив numpy. Некоторые из значений в этом массиве NaN
. Я хочу выполнить определенные операции с помощью этого массива. Например, рассмотрим массив:
[[ 0. 43. 67. 0. 38.]
[ 100. 86. 96. 100. 94.]
[ 76. 79. 83. 89. 56.]
[ 88. NaN 67. 89. 81.]
[ 94. 79. 67. 89. 69.]
[ 88. 79. 58. 72. 63.]
[ 76. 79. 71. 67. 56.]
[ 71. 71. NaN 56. 100.]]
Я пытаюсь взять каждую строку по одному, сортировать ее в обратном порядке, чтобы получить максимум 3 значения из строки и принять их среднее значение. Код, который я пробовал, это:
# nparr is a 2D numpy array
for entry in nparr:
sortedentry = sorted(entry, reverse=True)
highest_3_values = sortedentry[:3]
avg_highest_3 = float(sum(highest_3_values)) / 3
Это не работает для строк, содержащих NaN
. Мой вопрос в том, есть ли быстрый способ конвертировать все значения NaN
в ноль в массиве 2D numpy, чтобы у меня не было проблем с сортировкой и другими вещами, которые я пытаюсь сделать.