python flatten 2d list numpy