Как читать stdin в 2d массив целых чисел Python?

Я хотел бы прочитать двумерный массив целых чисел из стандартного ввода (или из файла) в Python.

Нерабочий код:

from StringIO import StringIO
from array import array

# fake stdin
stdin = StringIO("""1 2
3 4
5 6""")

a = array('i')
a.fromstring(stdin.read())

Это дает мне ошибку: a.fromstring (stdin.read ( )) ValueError: длина строки не кратна размеру элемента