Есть ли более чистый способ, чем использование для циклов

Question

Я хотел бы знать, есть ли какие-либо другие способы циклирования и управления данными, расположенными в отдельных массивах.

import numpy as np 

a = np.arange(2) 
b = np.arange(5) 
c = np.arange(5) 

l1 = [] 
for x in a: 
    l2 = [] 
    for y in b: 
     l3 = [] 
     y = x + 1 
     for z in c: 
      z = x + y 
      t = (x,y,z) 
      l3.append(t) 
     l2.append(l3) 
    l1.append(l2) 
print l1 

Answer 1

Этот код делает именно то, что вы делаете.

def method(lst, range1, range2): 
    for i in lst: 
     yield [[(i, i+1, 1+(i*2))]*range2]*range1 

Может даже быть превращен в выражение генератора:

def gen_method(lst, r1, r2): 
    return ([[(i, i+1, 1+(i*2))]*r2]*r1 for i in lst) 

проверить это самостоятельно, если вы хотите.

---

Мои тесты:

a = range(2) 
b = range(5) 
c = range(5) 

def your_method(a, b, c): 
    l1 = [] 
    for x in a: 
     l2 = [] 
     for y in b: 
      l3 = [] 
      y = x + 1 
      for z in c: 
       z = x + y 
       t = (x,y,z) 
       l3.append(t) 
      l2.append(l3) 
     l1.append(l2) 
    return l1 

def my_method(lst, range1, range2): 
    for i in lst: 
     yield [[(i, i+1, 1+(i*2))]*range2]*range1 

yours = your_method(a, b, c) 
mine = list(my_method(a, len(b), len(c))) 

print yours 
print '===' 
print mine 
print '===' 
print yours == mine 

>>> 
[[[(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)]], [[(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)]]] 
=== 
[[[(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)], [(0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1), (0, 1, 1)]], [[(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)], [(1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3), (1, 2, 3)]]] 
=== 
True 

Answer 2

Ну, вы можете сжимать код, используя списковых:

[[[(x,x+1,x*2 +1)]*len(c)]*len(b) for x in a] 

Что это делает петля для всех х а и создать список элементов, где каждый элемент является списком, сгенерированным для всех y в b, где каждый элемент содержит список: (x,x+1,2*x+1) для всех z в c.

Answer 3

Альтернативный подход заключается в использовании itertools, конвертировать полученный список numpy array и изменить его

import numpy as np 
import itertools as it 

a = np.arange(2) 
b = np.arange(5) 
c = np.arange(5) 

l1 = np.array([(i, i+1, i*2+1) for i, rb, rb in it.product(a, b, c)]) 
l1 = l1.reshape(len(a), len(b), len(c), len(d[0])) 

Это может быть, что это получает больше памяти, потребляющего, чем все другие подходы, по мере увеличения размера, но только две строки и создают уникальные элементы для каждого триплекса, а не просто создают несколько указателей на один и тот же объект.

Редактировать

Другим способом, что также позволяет сохранить триплет (я, я + 1, я * 1 + 1) в виде списка:

l1 = np.empty([len(a), len(b), len(c)], dtype=object) 
for i, rb, rb in it.product(a, b, c): 
    l1[i,ra, rb] = (i, i+1, i*2+1)