Create Fixed Size Array Python