需要提前下载好numpy模块。
下载示范:
2.输入cmd,进入命令行窗口
3.输入如下命令:
pip install numpy
包安装好之后,就可以开始正常写代码了
a = np.random.rand(10,3)
b = abs(a-0.5)
c = np.argsort(b)
注意到c数组中第一列的元素,表示的b中最小的元素在b中的下标,利用mask对其进行提取数据
mask = c[:,0]
for i in range(10):
print(a[i,mask[i]])
a = np.random.rand(10,3)
b = abs(a-0.5)
c = np.argsort(b)
mask = c[:,0]
for i in range(10):
print(a[i,mask[i]])