numpy的一些操作

原创于 2019-08-21 16:16:47 发布 · 102 阅读

CC 4.0 BY-SA版权

2 篇文章

订阅专栏

本文深入探讨了NumPy库中的meshgrid、where、sort、unique及cumsum等函数的使用技巧，通过实例演示如何生成网格数据，条件选择，数组排序，获取唯一值以及计算数组的累加和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

meshgrid函数接收两个一维数组，假设第一个参数为1×N，第二个为1×M；
- 那么得到两个结果为M×N的二维数组(注意不是N×M)
- 而第一个二维数组就是M行第一个一维数组
- 第二个二维数组就是N列第二个一维数组
np.where(cond,arr1,arr2)--------np.where(x>0,x,y)-----如果x的某个元素大于0,则取x,否则取y的该位元素
np.where(cond,scale1,scale2); np.where(arr>0,1,-1)-------如果arr的某个元素大于0，则该位赋值为1，否则赋值为-1；
arr.sort(axes_num):表示在arr的某一维度进行排序

 samples = np.random.normal(size=(4, 4))//伪随机正态分布的数组

arr = np.array([0, 1, 2, 3, 4, 5, 6, 7])  //计算累加和
arr.cumsum()    //对于多维数组可在函数中传入axes参数
Out: array([ 0,  1,  3,  6, 10, 15, 21, 28])