尝试这个:
import numpy as np
a = np.array([[12,45,50,60,30],
[12,34,5,6,88,-10,30,75],
[3,45,332,45,-12,-4,-64,12],
[12,45,3,22,323]], dtype='object')
for l in a:
for i in l:
if i<0:
l[l.index(i):] = [0] * len(l[l.index(i):])
a
output:
array([list([12, 45, 50, 60, 30]), list([12, 34, 5, 6, 88, 0, 0, 0]),
list([3, 45, 332, 45, 0, 0, 0, 0]), list([12, 45, 3, 22, 323])],
dtype=object)
第二种解决方案:
import numpy as np
def neg_to_zero(l):
for i in l:
if i<0:
l[l.index(i):] = [0] * len(l[l.index(i):])
a = np.array([[12,45,50,60,30],
[12,34,5,6,88,-10,30,75],
[3,45,332,45,-12,-4,-64,12],
[12,45,3,22,323]], dtype='object')
list(map(neg_to_zero, a))
a