如何用Python快速从已知数组派生出新数组的方法？

本文共计1105个文字，预计阅读时间需要5分钟。

要求描述：使用numpy进行数据分析时，常见的一个需求是根据已知数组生成新数组。这个问题可以分为两类：一是根据筛选条件生成子数组；二是根据变换条件生成新数组（新数组shape与原数组相同）。

需求描述

在利用numpy进行数据分析时，常有的一个需求是：根据已知的数组生成新数组。这个问题又可以分为两类：

1. 根据筛选条件生成子数组；
   
2. 根据变换条件生成新数组(新数组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的元素，然后生成一个新数组。