Как удалить все nan элементы из 3d numpy array?

Question

[Александр Ошеров]

Как удалить все nan элементы из 3d numpy array?
[ [[1,2,3],[4,5,np.nan],[5,6,7]],[[8,9,np.nan],[10,11,np.nan],[14,20,17]]

==>

[ [[1,2,3],[4,5],[5,6,7]], [[8,9],[10,11],[14,20,17]]

Comments

[U235U235]

А разве ваш результат будет валидным ndarray? Какой будет shape?
Можно развернуть массив в 1d, удалить nanы и собрать в какой-то 3d.
Заменить на нули, например, можно так:
a[np.where(np.isnan(a))]=0

[U235U235]

b=np.delete(a, np.where(np.isnan(a)))
Для плоского массива

[Александр Ошеров]

Для плоского это понятно

Answer 1

[Dimonchik]

а чем они мешают? они же nan