【Python3】Pythonのリスト内包表記【List comprehension】

0~5のリストを作る

以下のように記述すると、0～5のリストが作成できます。

x = [i for i in range(6)]
print(x)
>>> [0, 1, 2, 3, 4, 5]

x = [i+1000 for i in range(6)]
print(x)
>>> [1000, 1001, 1002, 1003, 1004, 1005]

names = ['佐藤', '鈴木', '渡辺']
x = [i + 'さん' for i in names]
print(x)
>>> ['佐藤さん', '鈴木さん', '渡辺さん']

条件付リスト内包表記

0~10までの数値の中で、偶数のみのリストの場合、以下のような記述になります。

x = [i for i in range(11) if i % 2 == 0]

「if i % 2 == 0」が条件です。これがTRUEのときだけ、新しいリストの要素になります。「if i % 2 == 0」は、iを2で割ったときのあまり（剰余）が0のときという意味です。

式の基本的な記述方法は以下のようになります。

[式 for 変数 in 繰り返しのオブジェクト if 条件式]

foods = ['apple', 'banana', 'lemon']
x = [i  for i in foods if 'a' in i]
print(x)
>>> ['apple', 'banana']

三項演算子のリスト内包表記

0~10までの数値のなかで、偶数だけ「ぐ」になるリストを作成します。

['ぐ' if i % 2 == else i for i in range(11)]

「’ぐ’ if % 2 == else」までが三項演算子です。

式の基本的な記述方法は以下のようになります。

[True時の値 if i 条件式 else False時の値 for 変数 in 繰り返しオブジェクト]

三項演算子のリスト内包表記を応用すると、以下のようになります。

x1 = [i for i in range(6)]
x2 = ['ぐ' if i % 2 == 0 else i for i in range(6)]
print(x1)
print(x2)
>>>> [0, 1, 2, 3, 4, 5]
>>>> ['ぐ', 1, 'ぐ', 3, 'ぐ', 5]

次回

Pythonで使用されるモジュール、パッケージの基本的な扱い方について解説します。

→【Python3】Pythonのモジュールの使い方【モジュール・パッケージ】