Kwadratura wszystkie elementy na liście

Question

Powiedziano mi do

Napisz funkcję, kwadrat (a), że pobiera tablicę, a, liczb i zwraca tablicę zawierającą każdej wartości a do kwadratu.

Początkowo miałem

def square(a): 
    for i in a: print i**2 

Ale to nie zadziała, ponieważ jestem drukowania, a nie powrót jakbym zapytał. Więc próbowałem

def square(a): 
     for i in a: return i**2 

ale to tylko kwadraty ostatni numer mojej tablicy. Jak mogę go ustawić na całej liście?

Answer 1

Można użyć wyrażeń listowych:

def square(list): 
    return [i ** 2 for i in list] 

Albo można go map:

def square(list): 
    return map(lambda x: x ** 2, list) 

Albo można użyć generatora. To nie zwróci listę, ale nadal można iterację nim, a ponieważ nie trzeba przeznaczyć całą nową listę, to prawdopodobnie bardziej efektywne niż przestrzeń innych opcji:

def square(list): 
    for i in list: 
     yield i ** 2 

Or można zrobić nudny stary for -loop, choć nie jest to tak idiomatyczne jak niektórzy programiści Python woleliby:

def square(list): 
    ret = [] 
    for i in list: 
     ret.append(i ** 2) 
    return ret 

Answer 2

Użyj listowych (jest to droga w czystym Pythonie):

>>> l = [1, 2, 3, 4] 
>>> [i**2 for i in l] 
[1, 4, 9, 16] 

Albo numpy (ugruntowane moduł):

>>> numpy.array([1, 2, 3, 4])**2 
array([ 1, 4, 9, 16]) 

W numpy, matematyka operacje na tablicach są domyślnie wykonywane w sposób elementarny. Dlatego możesz tam całą tablicę. **2

Inne możliwe rozwiązania byłyby oparte na map, ale w tym przypadku naprawdę sprawdziłbym listę. To Pythonic :) i rozwiązanie oparte na map, które wymaga lambda s is slower than LC.

Answer 3

def square(a): 
    squares = [] 
    for i in a: 
     squares.append(i**2) 
    return squares 

Answer 4

Używaj numpy.

import numpy as np 
b = list(np.array(a)**2) 

Answer 5

Jeszcze jedna mapa rozwiązanie:

def square(a): 
    return map(pow, a, [2]*len(a)) 

Answer 6

def square(a): 
    squares = [] 
    for i in a: 
     squares.append(i**2) 
    return squares 

tak jak zrobiłbym kwadratu liczb od 1-20 stosując powyższą funkcję

Answer 7

można zrobić

square_list =[i**2 for i in start_list] 

, która zwraca

[25, 9, 1, 4, 16] 

lub, jeśli lista ma już wartości

square_list.extend([i**2 for i in start_list]) 

która zwraca

[25, 9, 1, 4, 16] 

Uwaga: nie chcesz zrobić

square_list.append([i**2 for i in start_list]) 

jako zwraca listę na liście:

[[25, 9, 1, 4, 16]] 

Answer 8

import numpy as np 
a = [2 ,3, 4] 
np.square(a)