numpy、torch:[ ]里传函数,数组过滤

import numpy as np
lis = [1, 5, -9, 3, -6, 8, -7, 5, 5, 5, 4, 4, 4, 6, 6, 6, 6]
lis = np.array(lis)
lis[lis <= 0] = 100
print(lis)
[  1   5 100   3 100   8 100   5   5   5   4   4   4   6   6   6   6]
lis = [1, 5, -9, 3, -6, 8, -7, 5, 5, 5, 4, 4, 4, 6, 6, 6, 6]
lis = torch.tensor(lis)
lis[lis <= 0] = 100
print(lis)
tensor([  1,   5, 100,   3, 100,   8, 100,   5,   5,   5,   4,   4,   4,   6,
          6,   6,   6])
原文地址:https://www.cnblogs.com/DDBD/p/14097110.html