CODEWITHSUNDEEP
pythonlistinput
San
San

Reputation: 373

How to read multiple lists from STDIN in python?

I am newbie to python and trying to understand itertools.product. I am unable to read multiple lists from input.

Initially I have given manual input like below.

list1 = [1,2]
list2 = [3,4]
print(*product(list1, list2))

and got output as (1, 3) (1, 4) (2, 3) (2, 4), which is perfectly fine.

I wanted the same thing for multiple lists to be used in product function.

I have tried like below

TotList = product(list(map(int,input().split())) for _ in range(2)) #in range function 2 can be vary
for item in TotList:
    print(*item)

But it is not working like a product tool

Current input:

1 2
3 4

Output:

[1, 2]
[3, 4]

Expected output:

(1, 3) (1, 4) (2, 3) (2, 4)

Upvotes: 0

Views: 463

Answers (1)

Sunitha
Sunitha

Reputation: 12015

You have to specify * operator to unpack the iterables produced by map and feed it to product

>>> TotList = product(*(map(int,input().split()) for _ in range(2)))
1 2
3 4
>>> for item in TotList:
...     print(*item)
... 
1 3
1 4
2 3
2 4
>>>

Upvotes: 3

Related Questions