需求描述
在利用numpy进行数据分析时,常有的一个需求是:根据已知的数组生成新数组。这个问题又可以分为两类:
- 根据筛选条件生成子数组;
- 根据变换条件生成新数组(新数组shape与原数组相同)
下面简单总结.
生成子数组
情况1
已知数组a,以及若干筛选条件conds,要求从数组a中生成一个子数组b。
解决办法:b=a[conds]
。比如b=a[a>0]
,b=a[(a>=1)|(a<=-2)]
, b=a[(a>=1)&(a<=3)]
实例:如下
# 实例1.1:已知数组a,要求找出所有a>0的元素,然后生成一个新数组。
a = np.arange(-5,5,1)
print('原数组a:',a)
b = a[a>0]
print('实例1结果:',b)
# 实例1.2:已知数组a,要求找出所有a>=1或a<=-2的元素,然后生成一个新数组。
b = a[(a>=1) | (a<=-2)]
print('实例2结果:',b)
# 实例1.3:已知数组a,要求找出所有a>=1并且a<=3的元素,然后生成一个新数组。
b = a[(a>=1) & (a<=3)]
print(