error setting an array element with a sequence. python