Create A Boolean Array Python