Python: Compute the square of first N Fibonacci numbers, use map function and generate a list of the numbers

Python map: Exercise-10 with Solution

Write a Python program to compute the square of the first N Fibonacci numbers, using the map function and generate a list of the numbers.

Fibonacci numbers:

From Wikipedia,
In mathematics, the Fibonacci numbers, commonly denoted Fn, form a sequence, called the Fibonacci sequence, such that each number is the sum of the two preceding ones, starting from 0 and 1. That is,

F₀ = 0,   F₁ = 1,

and

F_n = F_{n - 1} + F_{n - 2},

for n > 1.

The beginning of the sequence is thus:

0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144,....

Sample Solution:

Python Code :

import itertools
n = 10
def fibonacci_nums(x=0, y=1):
    yield x
    while True:
        yield y
        x, y = y, x + y
print("First 10 Fibonacci numbers:")
result = list(itertools.islice(fibonacci_nums(), n))
print(result)
square = lambda x: x * x 
print("\nAfter squaring said numbers of the list:")
print(list(map(square, result)))

Sample Output:

First 10 Fibonacci numbers:
[0, 1, 1, 2, 3, 5, 8, 13, 21, 34]

After squaring said numbers of the list:
[0, 1, 1, 4, 9, 25, 64, 169, 441, 1156]

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Python program to create a new list taking specific elements from a tuple and convert a string value to integer.
Next: Write a Python program to compute the sum of elements of a given array of integers, use map() function.