numpy模块（2）

1、利用布尔值来取元素 import numpy as np mask = np.array([1,0,1],dtype=bool) # 1表示取对应的元素，0表示不取 arr = np.array([[1,2,3],[4,5,6],[7,8,9]]) print(arr[mask,2]) print(arr[:,mask]) 运行结果： [3 9] [[1 3] [4 6] [7 9]] 一

>>阅读原文<<