web-dev-qa-db-fra.com

Tableau répété avec transformation

Comment puis-je répéter la séquence d'exemple suivante:

l = np.array([3,4,5,6,7])

Jusqu'à n fois, en doublant les valeurs à chaque répétition. Donc pour n=3:

[3, 4, 5, 6, 7, 6,  8, 10, 12, 14, 12, 16, 20, 24, 28]

Existe-t-il un moyen simple d'éviter les boucles avec numpy peut-être?

13
yatu

numpy.outer + numpy.ndarray.ravel :

>>> a = np.array([3,4,5,6,7])                                                                                          
>>> n = 3                                                                                                              
>>> factors = 2**np.arange(n)                                                                                          
>>> np.outer(factors, a).ravel()                                                                                       
array([ 3,  4,  5,  6,  7,  6,  8, 10, 12, 14, 12, 16, 20, 24, 28])

Détails:

>>> factors                                                                                                            
array([1, 2, 4])
>>> np.outer(factors, a)                                                                                               
array([[ 3,  4,  5,  6,  7], # 1*a
       [ 6,  8, 10, 12, 14], # 2*a
       [12, 16, 20, 24, 28]]) # 4*a
10
timgeb

Vous pouvez utiliser la compréhension par concaténation et liste en utilisant des puissances de 2 comme facteur multiplicatif. Ici, 3 est le nombre de répétitions dont vous avez besoin.

l = np.array([3,4,5,6,7])
final = np.concatenate([l*2**(i) for i in range(3)])
print (final)

array([ 3,  4,  5,  6,  7,  6,  8, 10, 12, 14, 12, 16, 20, 24, 28])
3
Bazingaa

Voici un moyen:

>>> (l * [[1], [2], [4]]).flatten()
array([ 3,  4,  5,  6,  7,  6,  8, 10, 12, 14, 12, 16, 20, 24, 28])

(Ne fonctionne que si vous voulez des multiples de l'entrée.)

2
NPE

Vous pouvez faire une boucle sur n et concaténer une nouvelle liste à chaque fois:

l = [3,4,5,6,7]
n = 3
new_l = []
for i in range(n):
    new_l += [j*2**i for j in l]
new_l = np.array(new_l)

Sortie

array([3, 4, 5, 6, 7, 6, 8, 10, 12, 14, 12, 16, 20, 24, 28])
1
Tim

Vous pourriez faire:

import numpy as np

l = np.array([3, 4, 5, 6, 7])

rows = np.tile(l, 3).reshape(-1, len(l)) * np.power(2, np.arange(3)).reshape(-1, 1)
print(rows.flatten())

Sortie

[ 3  4  5  6  7  6  8 10 12 14 12 16 20 24 28]
1
Daniel Mesejo

Vous pouvez utiliser np.power avec np.ndarray.ravel :

A = np.array([3,4,5,6,7])

res = (A * np.power(2, np.arange(3))[:, None]).ravel()

print(res)

array([ 3,  4,  5,  6,  7,  6,  8, 10, 12, 14, 12, 16, 20, 24, 28])
1
jpp

Vous pouvez le faire en python pur en utilisant une compréhension de liste:

l = [3,4,5,6,7]

n     = 3
mult  = [1 * 2**i for i in range(n)]
final = list([x*m for m in mult for x in l])
0
feebarscevicius