MST

星途 面试题库

面试题:Python列表嵌套元素的修改技巧

有一个嵌套列表nested_list = [[1, 2], [3, 4], [5, 6]],请编写代码将内层列表中所有大于3的元素替换为其平方值,阐述实现思路并给出代码。
39.0万 热度难度
编程语言Python

知识考点

AI 面试

面试题答案

一键面试

实现思路

  1. 遍历外层列表,对于每一个内层列表。
  2. 遍历内层列表中的每个元素。
  3. 判断元素是否大于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)