2009年12月23日水曜日

リスト内包表記

for文でlistを生成する場合に、リスト内包表記が使えます。例えば、

li = []
for i in [1, 2, 3, 4]:
    j = i ** i # iのi乗
    li.append(j)

普通は上のように書くところを、

li = [i ** i for i in [1, 2, 3, 4]]

と書くことが出来ます。さらに

li = []
for i in [1, 2, 3, 4]:
    if not i % 2 # 偶数
        j = i ** i
        li.append(j)

のようにif文がある場合には、

li = [i ** i for i in [1, 2, 3, 4] if not i % 2]

と書くことが出来ます。

0 件のコメント:

コメントを投稿