Numpy tricks

最新推荐文章于 2020-12-14 21:22:50 发布

aureole420

最新推荐文章于 2020-12-14 21:22:50 发布

阅读量100

点赞数

本文链接：https://blog.csdn.net/aureole420/article/details/90287343

版权

本文介绍了Numpy的一些实用技巧，包括如何根据排序获取2D矩阵的索引，利用numpy的broadcasting进行不同尺寸矩阵运算，以及如何使用Boolean数组进行K折交叉验证。特别强调了numpy的argsort函数在获取排序索引时的重要性，以及在避免内存消耗方面的方法。

摘要由CSDN通过智能技术生成

应用场景：KNN中给定dataset X, Y, 在X中寻找dist最小的那几个的index然后到Y中去获取label。

Given a 2D matrix: data = np.array([[3,1],[2,4]])

排序
- 直接 np.sort(data, axis = none) generate a 1D array [1,2,3,4],
- np.sort if assign axis, e.g. np.sort(data, axis=0) will generate a SAME DIMENSION matrix sorted in given axis. e.g [[2,1],[3,4]] —same holds for HIGH DIMENSION.
排序，但要求顺序给出index .e.g. 1D matrix x=np.array([2,1,3]), 要给出index ordered_mast = [1,0,2] so x[ordered_mask] = [1,2,3].
- Use np.argsort(x) https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.ht

###2. BroadCasting

numpy broadcasting真是tricky！

定义用来match different-dim matrix h

关注