flatten 2d list python numpy