Simplified logic to create ndarray in Poisson predict