我有一个数组,string
如:
[['2', '1', '1', '1', '4'] ['4', '3', '3', '5', '6'] ['7', '7', '2', '2', '7']]
我想得到的结果与integer
元素为:
[[0,0,0,0,0] [0,0,0,1,1] [1,1,0,0,1]]
基于以下规则:如果元素大于4,则将其替换为1
else0
我试图在Python中使用双循环,但是它太复杂了,有没有简单的方法可以完成这项工作?
>>> L = [['2', '1', '1', '1', '4'], ['4', '3', '3', '5', '6'], ['7', '7', '2', '2', '7']] >>> [[(1 if int(num)>4 else 0) for num in row] for row in L] [[0, 0, 0, 0, 0], [0, 0, 0, 1, 1], [1, 1, 0, 0, 1]]