python 列表推导式 - 还看不懂列表推导式？来看看这篇，说不定对你有帮助

本文主要是将列表推导式和for循环语句进行对比，来使列表推导式更加容易理解

单层for循环

单层for循环 - 只有for循环

代码

# 单层for循环-只有for循环

# 初始化列表
list_aa = list()
list_ab = list()

# 使用普通的for循环
for i in range(10):
    list_aa.append(i)
print(f'list_aa={list_aa}')

for i in range(10):
    list_ab.append(i * i)
print(f'list_ab={list_ab}')

# 使用列表推导式
list_ac = [i for i in range(10)]
print(f'list_ac={list_ac}')

list_ad = [i * i for i in range(10)]
print(f'list_ad={list_ad}')

结果

list_aa=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
list_ab=[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
list_ac=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
list_ad=[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

单层for循环 - 存在if嵌套

代码

# 单层for循环 - 存在if嵌套

# 初始化列表
list_ba = list()
list_bb = list()

# 使用普通的for循环
for i in range(10):
    if i % 3 == 0:
        list_ba.append(i)
print(f'list_ba={list_ba}')

for i in range(10):
    if i % 3 == 0:
        list_bb.append(i * i)
print(f'list_bb={list_bb}')

# 使用列表推导式
list_bc = [i for i in range(10) if i % 3 == 0]
print(f'list_bc={list_bc}')

list_bd = [i * i for i in range(10) if i % 3 == 0]
print(f'list_bd={list_bd}')

结果

list_ba=[0, 3, 6, 9]
list_bb=[0, 9, 36, 81]
list_bc=[0, 3, 6, 9]
list_bd=[0, 9, 36, 81]

单层for循环 - 存在if-else嵌套

代码

# 单层for循环 - 存在if-else嵌套

# 初始化列表
list_ca = list()
list_cb = list()

# 使用普通的for循环
for i in range(10):
    if i % 3 == 0:
        list_ca.append(i)
    else:
        list_ca.append(-i)
print(f'list_ca={list_ca}')

for i in range(10):
    if i % 3 == 0:
        list_cb.append(i * i)
    else:
        list_cb.append(i * i * i)
print(f'list_cb={list_cb}')

# 使用列表推导式
list_cc = [i if i % 3 == 0 else -i for i in range(10)]
print(f'list_cc={list_cc}')

list_cd = [i * i if i % 3 == 0 else i * i * i for i in range(10)]
print(f'list_cd={list_cd}')

结果

list_ca=[0, -1, -2, 3, -4, -5, 6, -7, -8, 9]
list_cb=[0, 1, 8, 9, 64, 125, 36, 343, 512, 81]
list_cc=[0, -1, -2, 3, -4, -5, 6, -7, -8, 9]
list_cd=[0, 1, 8, 9, 64, 125, 36, 343, 512, 81]

单层for循环 - 存在多层if-else嵌套

代码

# 单层for循环 - 存在多层if-else嵌套

# 初始化列表
list_da = list()

# 使用普通的for循环
for i in range(10):
    if i % 3 == 0:
        list_da.append(i * i)
    else:
        if i % 5 == 0:
            list_da.append(i * i * i)
        else:
            list_da.append(i * i * i * i)
print(f'list_da={list_da}')

# 使用列表推导式
list_db = [i * i if i % 3 == 0 else i * i * i if i % 5 == 0 else i * i * i * i for i in range(10)]
print(f'list_db={list_db}')

结果

list_da=[0, 1, 16, 9, 256, 125, 36, 2401, 4096, 81]
list_db=[0, 1, 16, 9, 256, 125, 36, 2401, 4096, 81]

多层for循环

多层for循环 - 只有for循环

代码

# 初始化列表
my_list_multilayer = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
list_ea = list()

# 使用普通的for循环
for i in my_list_multilayer:
    for j in i:
        list_ea.append(j)
print(f'list_ea={list_ea}')

# 使用列表推导式
list_eb = [j for i in my_list_multilayer for j in i]
print(f'list_eb={list_eb}')

结果

list_ea=[1, 2, 3, 4, 5, 6, 7, 8, 9]
list_eb=[1, 2, 3, 4, 5, 6, 7, 8, 9]

全部代码

代码

# -*- coding:utf-8 -*-
# FileName: list_comprehension
# Author: Bamboo
# CreateTime: 2022/7/26 16:14

# =====================================
# 单层for循环
# =====================================

# 单层for循环-只有for循环

# 初始化列表
list_aa = list()
list_ab = list()

# 使用普通的for循环
for i in range(10):
    list_aa.append(i)
print(f'list_aa={list_aa}')

for i in range(10):
    list_ab.append(i * i)
print(f'list_ab={list_ab}')

# 使用列表推导式
list_ac = [i for i in range(10)]
print(f'list_ac={list_ac}')

list_ad = [i * i for i in range(10)]
print(f'list_ad={list_ad}')

# 单层for循环 - 存在if嵌套

# 初始化列表
list_ba = list()
list_bb = list()

# 使用普通的for循环
for i in range(10):
    if i % 3 == 0:
        list_ba.append(i)
print(f'list_ba={list_ba}')

for i in range(10):
    if i % 3 == 0:
        list_bb.append(i * i)
print(f'list_bb={list_bb}')

# 使用列表推导式
list_bc = [i for i in range(10) if i % 3 == 0]
print(f'list_bc={list_bc}')

list_bd = [i * i for i in range(10) if i % 3 == 0]
print(f'list_bd={list_bd}')

# 单层for循环 - 存在if-else嵌套

# 初始化列表
list_ca = list()
list_cb = list()

# 使用普通的for循环
for i in range(10):
    if i % 3 == 0:
        list_ca.append(i)
    else:
        list_ca.append(-i)
print(f'list_ca={list_ca}')

for i in range(10):
    if i % 3 == 0:
        list_cb.append(i * i)
    else:
        list_cb.append(i * i * i)
print(f'list_cb={list_cb}')

# 使用列表推导式
list_cc = [i if i % 3 == 0 else -i for i in range(10)]
print(f'list_cc={list_cc}')

list_cd = [i * i if i % 3 == 0 else i * i * i for i in range(10)]
print(f'list_cd={list_cd}')

# 单层for循环 - 存在多层if-else嵌套

# 初始化列表
list_da = list()

# 使用普通的for循环
for i in range(10):
    if i % 3 == 0:
        list_da.append(i * i)
    else:
        if i % 5 == 0:
            list_da.append(i * i * i)
        else:
            list_da.append(i * i * i * i)
print(f'list_da={list_da}')

# 使用列表推导式
list_db = [i * i if i % 3 == 0 else i * i * i if i % 5 == 0 else i * i * i * i for i in range(10)]
print(f'list_db={list_db}')

# =====================================
# 多层for循环
# =====================================

# 初始化列表
my_list_multilayer = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
list_ea = list()

# 使用普通的for循环
for i in my_list_multilayer:
    for j in i:
        list_ea.append(j)
print(f'list_ea={list_ea}')

# 使用列表推导式
list_eb = [j for i in my_list_multilayer for j in i]
print(f'list_eb={list_eb}')

结果

list_aa=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
list_ab=[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
list_ac=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
list_ad=[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
list_ba=[0, 3, 6, 9]
list_bb=[0, 9, 36, 81]
list_bc=[0, 3, 6, 9]
list_bd=[0, 9, 36, 81]
list_ca=[0, -1, -2, 3, -4, -5, 6, -7, -8, 9]
list_cb=[0, 1, 8, 9, 64, 125, 36, 343, 512, 81]
list_cc=[0, -1, -2, 3, -4, -5, 6, -7, -8, 9]
list_cd=[0, 1, 8, 9, 64, 125, 36, 343, 512, 81]
list_da=[0, 1, 16, 9, 256, 125, 36, 2401, 4096, 81]
list_db=[0, 1, 16, 9, 256, 125, 36, 2401, 4096, 81]
list_ea=[1, 2, 3, 4, 5, 6, 7, 8, 9]
list_eb=[1, 2, 3, 4, 5, 6, 7, 8, 9]