New Array Within An Array Python Size Function