pythonでfor文をネストしたいときは、代わりにlist内包表記やmap関数を使う方が、コードが短く計算も速い

In [1]:

import numpy as np

In [2]:

# リスト内包表記でリストを作る
lis1 = [x for x in range(5)]

In [3]:

lis1

Out[3]:

[0, 1, 2, 3, 4]

In [4]:

# リスト内包表記で行列を作る
lis2 = [[x + y for x in range(5)] for y in range(5)]

In [5]:

lis2

Out[5]:

[[0, 1, 2, 3, 4],
 [1, 2, 3, 4, 5],
 [2, 3, 4, 5, 6],
 [3, 4, 5, 6, 7],
 [4, 5, 6, 7, 8]]

In [6]:

# リスト内包表記で2成分リストの行列を作る
lab1 = [1, 2, 3]
lab2 = [4, 5, 6]
lis3 = np.array([[[x, y] for x in lab1] for y in lab2])

In [7]:

lis3

Out[7]:

array([[[1, 4],
        [2, 4],
        [3, 4]],

       [[1, 5],
        [2, 5],
        [3, 5]],

       [[1, 6],
        [2, 6],
        [3, 6]]])

In [8]:

# map関数とlambda関数で4成分リストの行列を作る
lis3t = np.array(lis3).T
lis4 = np.array(list(map(lambda x, y: [x, y, x, y], *lis3t))).transpose(2, 0, 1)

In [9]:

lis4

Out[9]:

array([[[1, 4, 1, 4],
        [2, 4, 2, 4],
        [3, 4, 3, 4]],

       [[1, 5, 1, 5],
        [2, 5, 2, 5],
        [3, 5, 3, 5]],

       [[1, 6, 1, 6],
        [2, 6, 2, 6],
        [3, 6, 3, 6]]])

In [10]:

# map関数とlambda関数で1成分の行列を作る
lis5 = np.array(list(map(lambda x, y: x, *lis3t))).T

In [11]:

lis5

Out[11]:

array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

In [ ]: