Bu iki satırlı kodu nasıl hızlandırabilirim?

Question

Ben aşağıdaki kodu hızlandırmak gerekir:

for i in range(0, 2**N): 

    output[i] = f(np.array(map(int, bin(i)[2:].zfill(N)))) 

N civarında 30, bu nedenle kodu (Dizüstü bilgisayarımda yaklaşık 33 saat sürer) çok yavaştır. f() işlevinin argümanı, i dizininin ikili gösterimidir ve f(), rasgele bir vektörleştirilebilir işlev olabilir. Ben uzman değilim, ama kodu hızlandırmak içinargümanını vectorize etmem gerektiği anlamına gelen for döngüsünden kurtulmayı düşünüyordum. Başka bir deyişle, 0'dan 2**N'a kadar sayıların ikili gösterimleriyle bir matris oluşturmam gerekiyor. Bu aşağıdaki kod üzerinden elde edilebilir:

list(itertools.product([0, 1], repeat=N)) 

Ben this link de bulunduğunu bildirdi. Ancak, bana itertools'un çok yavaş olduğu ve 2**30'un yaklaşık bir milyar olduğu için çok fazla bellek gerektirdiği görülüyor.

Bu kodu daha hızlı yapmak için herhangi bir öneriniz var mı? Şimdiden teşekkürler.

Answer 1

daima profili: öyle değil dizeleri ile uğraşmak zorundadır çünkü

>>> timeit.timeit("for i in range(0, 2**N): numpy.array(map(int, bin(i)[2:].zfill(N)))", "import numpy; N=5", number=100000) 
26.472519159317017 
>>> timeit.timeit("for t in itertools.product((0, 1), repeat=N): numpy.array(t)", "import numpy, itertools; N=5", number=100000) 
6.129688024520874 

Sen itertools.product yöntem oldukça hızlı olduğunu görebilirsiniz.

Sorun çoğu zaman f işlevinde harcanmış olabilir.

Başka bir çözüm, f bir tamsayı kabul etmek ve bir ikili alan olarak kullanmak olabilir.