面试题答案
一键面试实现思路
- 遍历外层列表,对于每一个内层列表。
- 遍历内层列表中的每个元素。
- 判断元素是否大于3,如果是,则将其替换为该元素的平方值。
代码
nested_list = [[1, 2], [3, 4], [5, 6]]
for sublist in nested_list:
for i in range(len(sublist)):
if sublist[i] > 3:
sublist[i] = sublist[i] ** 2
print(nested_list)