แทนที่ค่าลบในอาร์เรย์จำนวนนับ

มีวิธีง่ายๆในการแทนที่ค่าลบทั้งหมดในอาร์เรย์ด้วย 0 หรือไม่?

ฉันมีบล็อกที่สมบูรณ์เกี่ยวกับวิธีการใช้อาร์เรย์ NumPy

เช่น

a = array([1, 2, 3, -4, 5])

ฉันต้องกลับ

[1, 2, 3, 0, 5]

a < 0 ให้:

[False, False, False, True, False]

นี่คือจุดที่ฉันติดอยู่ - วิธีใช้อาร์เรย์นี้เพื่อแก้ไขอาร์เรย์ดั้งเดิม

คุณอยู่ครึ่งทางแล้ว ลอง:

In [4]: a[a < 0] = 0

In [5]: a
Out[5]: array([1, 2, 3, 0, 5])

ลองnumpy.clip:

>>> import numpy
>>> a = numpy.arange(-10, 10)
>>> a
array([-10,  -9,  -8,  -7,  -6,  -5,  -4,  -3,  -2,  -1,   0,   1,   2,
         3,   4,   5,   6,   7,   8,   9])
>>> a.clip(0, 10)
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

clip(0)คุณสามารถคลิปเพียงครึ่งล่างด้วย

>>> a = numpy.array([1, 2, 3, -4, 5])
>>> a.clip(0)
array([1, 2, 3, 0, 5])

clip(max=n)คุณสามารถคลิปเพียงครึ่งบนกับ (นี่ดีกว่าคำแนะนำก่อนหน้าของฉันมากซึ่งเกี่ยวข้องกับการส่งผ่านNaNไปยังพารามิเตอร์แรกและใช้outเพื่อบังคับประเภท):

>>> a.clip(max=2)
array([ 1,  2,  2, -4,  2])

อีกแนวทางที่น่าสนใจคือการใช้where:

>>> numpy.where(a <= 2, a, 2)
array([ 1,  2,  2, -4,  2])

สุดท้ายพิจารณาคำตอบของaix ฉันชอบclipการดำเนินการที่เรียบง่ายเนื่องจากเป็นการจัดทำเอกสารด้วยตนเอง แต่คำตอบของเขาเหมาะสำหรับการดำเนินการที่ซับซ้อนมากกว่า

อีกหนึ่งโซลูชัน Python ที่เรียบง่ายโดยไม่ต้องใช้ numpy:

[0 if i < 0 else i for i in a]

ไม่จำเป็นต้องกำหนดฟังก์ชันเสริมใด ๆ

a = [1, 2, 3, -4, -5.23, 6]
[0 if i < 0 else i for i in a]

ผลตอบแทน:

[1, 2, 3, 0, 0, 6]

และความเป็นไปได้อื่น:

In [2]: a = array([1, 2, 3, -4, 5])

In [3]: where(a<0, 0, a)
Out[3]: array([1, 2, 3, 0, 5])

นี่คือวิธีทำใน Python โดยไม่มี NumPy สร้างฟังก์ชันที่ส่งคืนสิ่งที่คุณต้องการและใช้การทำความเข้าใจรายการหรือฟังก์ชันแผนที่

>>> a = [1, 2, 3, -4, 5]

>>> def zero_if_negative(x):
...   if x < 0:
...     return 0
...   return x
...

>>> [zero_if_negative(x) for x in a]
[1, 2, 3, 0, 5]

>>> map(zero_if_negative, a)
[1, 2, 3, 0, 5]