根据Python中的规则将所有字符串更改为整数

Question

我有一个数组，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，则将其替换为1else0

我试图在Python中使用双循环，但是它太复杂了，有没有简单的方法可以完成这项工作？

Answer 1

>>> 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]]