Python のリスト内包表記を簡単に理解する

簡単に言えば、リスト内包表記は、リストを作成するためのより便利な方法を提供します。つまり、元の数行のコードを 1 行に置き換えることができます。その構造は次のとおりです。

newlist = [expression for member in iterable (if conditional)]

変換すると、反復可能なオブジェクト iterable の場合、それが条件条件を満たす場合 (条件がない場合は、直接 True であることと同等)、毎回そこからメンバーを取り出し、それを式関数に入れます。処理し、それを新しいリストの中央に追加します。

外側の角括弧が削除された場合、内包表記は基本的にリストではなくジェネレーター オブジェクトを返すことに注意してください。

print(type(i for i in range(1, 100)))
# <class 'generator'>

print(type([i for i in range(1, 100)]))
# <class 'list'>

---

例1

1 から 99 までの整数配列を作成するとします。元の書き込み方法は次のとおりです。

ret = []
for i in range(1, 100):
	ret.append(i)
print(ret)

リスト内包表記は次のように記述されます。

ret = [i for i in range(1, 100)]
print(ret)

---

例 2

1 から 99 までの奇数の配列を作成するとします。元の書き方は次のとおりです。

ret = []
for i in range(1, 100):
	if i % 2 == 1:
		ret.append(i)
print(ret)

リスト内包表記は次のように記述されます。

ret = [i for i in range(1, 100) if i % 2 == 1]
print(ret)

---

例 3

1 から 9 までの平方数を含む配列を作成するとします。本来の書き方は次のとおりです。

ret = []
for i in range(1, 10):
	ret.append(i ** 2)
print(ret)

リスト内包表記は次のように記述されます。

ret = [i ** 2 for i in range(1, 10)]
print(ret)

---

例 4

要素 K を配列から削除すると仮定すると、本来の書き方は次のようになります。

lst = [1, 2, 3, 4, 5, 3, 6]
K = 3

ret = []
for x in lst:
    if x != K:
        ret.append(x)
print(ret)

リスト内包表記は次のように記述されます。

lst = [1, 2, 3, 4, 5, 3, 6]
K = 3

ret = [x for x in lst if x != K]
print(ret)

ここから、リストの内包表記はフィルター関数の置き換えと同等であることがわかります (さらに簡潔になる可能性もあります)。フィルタの使用方法は次のとおりです。

lst = [1, 2, 3, 4, 5, 3, 6]
K = 3

ret = list(filter(lambda x: x != K, lst))
print(ret)