Python神技：用列表推导式让你的代码更优雅

Python神技：用列表推导式让你的代码更优雅

如果你在Python代码中经常看到一行简洁的for循环，并能生成一个完整的列表，那么你遇到的就是“列表推导式”。它不仅是Pythonic编程的典范，更能显著提升代码的可读性和执行效率。

什么是列表推导式？

简单来说，它是一种用简洁的语法快速创建列表的方法。

传统方式 vs. 列表推导式

假设我们需要创建一个包含0到9所有数字平方的列表。

• 传统for循环写法：

  squares = []
  for i in range(10):
      squares.append(i**2)
  print(squares) # 输出: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
  

• 列表推导式写法：

  squares = [i**2 for i in range(10)]
  print(squares) # 输出: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
  

对比之下，推导式将四行代码压缩为一行，意图清晰，一目了然。

进阶用法：添加条件过滤

推导式的强大之处在于可以方便地加入条件判断。

例如，我们只想得到偶数的平方：

even_squares = [i**2 for i in range(10) if i % 2 == 0]
print(even_squares) # 输出: [0, 4, 16, 36, 64]

你甚至可以使用if-else三元表达式：

results = [x if x % 2 == 0 else 'odd' for x in range(5)]
print(results) # 输出: [0, 'odd', 2, 'odd', 4]

不仅仅是列表

推导式的思想可以延伸到其他数据结构：

• 字典推导式： {key: value for key, value in some_iterable}
• 集合推导式： {x for x in range(5)} (自动去重)

总结与提醒

列表推导式是Python送给开发者的“语法糖”，它让代码更简洁、更高效。但切记，过长的或嵌套过多的推导式会降低可读性。在追求简洁的同时，务必保证代码清晰易懂。

下次当你准备写for循环来构建列表时，不妨先想想：能否用一行优雅的推导式来解决？