Làm thế nào để cắt bớt các giá trị của một mảng có khối lượng 2D

Question

Tôi có mảng numpy hai chiều (uint16), làm cách nào tôi có thể cắt bớt tất cả các giá trị trên một rào cản nhất định (nói 255) về rào cản đó? Các giá trị khác phải giữ nguyên. Sử dụng một vòng lặp lồng nhau có vẻ là không hiệu quả và vụng về.

Answer 1

import numpy as np 
my_array = np.array([[100, 200], [300, 400]],np.uint16) 
my_array[my_array > 255] = 255 

sản lượng sẽ được

array([[100, 200], 
     [255, 255]], dtype=uint16) 

Answer 2

Trong trường hợp câu hỏi của bạn không phải là liên quan đến độ sâu bit là câu trả lời JBernardo của, cách tổng quát hơn để làm điều đó sẽ là một cái gì đó như: (sau chỉnh sửa, câu trả lời của tôi bây giờ là khá nhiều giống như mình)

 
def trunc_to(my_array, limit): 
    too_high = my_array > limit 
    my_array[too_high] = limit 

Here 'sa liên kết giới thiệu tốt đẹp cho numpy bool lập chỉ mục.

Answer 3

thực sự có một phương pháp cụ thể cho điều này, 'kẹp':

import numpy as np 
my_array = np.array([[100, 200], [300, 400]],np.uint16) 
my_array.clip(0,255) # clip(min, max) 

đầu ra:

array([[100, 200], 
     [255, 255]], dtype=uint16)