seuil en 2D tableau numpy

J'ai un tableau de la forme 512x512 qui contient des nombres entre 0 et 100 au i ' eme et la j-ème position. Maintenant, je veux sélectionner tableau[i,j] < 25 et zéro à d'autres endroits. J'ai essayé avec array = array[where(array<25)], ce qui me donne un tableau 1D, mais je veux 2D. Merci de m'aider à résoudre ce.

Double Possible de Définir un tableau numpy éléments à zéro si ils sont en dessous d'un seuil spécifique

OriginalL'auteur physics_for_all | 2016-04-19

  1. 17

    Une solution:

    result = (array < 25) * array

    La première partie array < 25 vous donne un tableau de la même forme, c'est-à 1 (Vrai), où les valeurs sont à moins de 25 et 0 (Faux) sinon. L'élément de sage multiplication avec le tableau d'origine conserve les valeurs qui sont inférieures à 25 et définit le reste à 0. Cela ne change pas le tableau d'origine

    Une autre possibilité est de définir toutes les valeurs qui sont >= 25 à zéro dans le tableau d'origine:

    array[array >= 25] = 0

    OriginalL'auteur kazemakase