python由已知数组快速生成新数组的方法

需求描述

在利用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的元素，然后生成一个新数组。
b = a[(a>=1) & (a<=3)]
print('实例3结果：',b)

运行结果：

原数组a： [-5 -4 -3 -2 -1  0  1  2  3  4]
实例1结果： [1 2 3 4]
实例2结果： [-5 -4 -3 -2  1  2  3  4]
实例3结果： [1 2 3]

情况2

已知数组a和数组b（shape相同），以及对数组a的筛选条件conds_a。要求从数组b中生成一个子数组c，其中的元素id，与满足筛选条件的数组a的元素id一一对应。

解决办法：

c = b[conds_a],比如c = b[(a>=1) & (a<=3)]

实例：如下

a = np.arange(-5,5,1)
b = np.arange(-50,50,10)
print('数组a:',a)
print('数组b:',b)

c = b[(a>=1) & (a<=3)]
print('新数组c:', c)

运行结果：

数组a: [-5 -4 -3 -2 -1  0  1  2  3  4]
数组b: [-50 -40 -30 -20 -10   0  10  20  30  40]
新数组c: [10 20 30]

变换成新数组

已知数组a，以及若干变换条件conds，要求生成一个新数组b（与原数组shape相同）。解决办法：

方法1：np.where(where(condition, [x, y]))
使用场景：当变换条件只有两个以下时，比如实例2.1。该方法等价于:if condition x else y

方法2： np.select(condlist, choicelist, default=0)
使用场景：当变换条件有任意多个时，比如实例2.2。该方法等价于:

if condlist[0]:
  choicelist[0]
elif condilist[1]:
  choicelist[1]
...
else:
  default

方法3：np.piecewise(x, condlist, funclist, *args, **kw)

使用场景：同方法2，只不过变换条件较复杂，无法直接写出，要用函数来表示。
实例：如下

# 实例2.1：已知数组a，要求对所有a<0的元素取绝对值，对其他元素设为0，然后生成一个新数组
a = np.arange(-5,5,1)
print(a)
b = np.where(a<0, abs(a),0)
print('实例2.1结果：',b)

# 实例2.2：已知数组a，要求对所有a<0的元素取绝对值，对a=0的元素+100，对a>0的元素平方，然后生成一个新数组
b = np.select([a<0, a==0, a>0], [abs(a), a+100, a**2])
print('实例2.2结果：',b)

运行结果：

[-5 -4 -3 -2 -1  0  1  2  3  4]
实例2.1结果： [5 4 3 2 1 0 0 0 0 0]
实例2.2结果： [5 4 3 2 1 100 1 4 916]

到此这篇关于python由已知数组快速生成新数组的方法的文章就介绍到这了,更多相关python 已知数组快速生成新数组内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们

赞 (0)

相关推荐

  • Python numpy.array()生成相同元素数组的示例

    如下所示: new_array = np.zeros((5,4)) for i in range(3): new_array[i] = np.array([0.25]*4) 运行结果: >>> new_array array([[0.25, 0.25, 0.25, 0.25], [0.25, 0.25, 0.25, 0.25], [0.25, 0.25, 0.25, 0.25], [0. , 0. , 0. , 0. ], [0. , 0. , 0. , 0. ]]) 以上这篇Pytho

  • python二维键值数组生成转json的例子

    今天出于需要,要将爬虫爬取的一些数据整理成二维数组,再编码成json字符串传入数据库 那么问题就来了,在php中这个过程很简便 ,类似这样: $arr[$key1][$key2]=$value 在python中,字符串竟然不能当做键值,我本身也并不了解python,查了资料也没解决,后来找到这个函数: dict.setdefault(key, default=None) 和get()类似, 但如果键不存在于字典中,将会添加键并将值设为default 于是乎,将键值当做两个参数传进去就等实现类似的

  • python3.x 生成3维随机数组实例

    如下所示: import numpy as np a=np.random.randint(0,10,size=[3,3,3]) print(a) 以上这篇python3.x 生成3维随机数组实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我们.

  • Python生成随机数组的方法小结

    Python生成随机数组的方法小结

    本文实例讲述了Python生成随机数组的方法.分享给大家供大家参考,具体如下: 研究排序问题的时候常常需要生成随机数组来验证自己排序算法的正确性和性能,今天把Python生成随机数组的方法稍作总结,以备以后查看使用. 一.使用random模块生成随机数组 python的random模块中有一些生成随机数字的方法,例如random.randint, random.random, random.uniform, random.randrange,这些函数大同小异,均是在返回指定范围内的一个整数或浮点

  • Python动态生成多维数组的方法示例

    Python动态生成多维数组的方法示例

    本文实例讲述了Python动态生成多维数组的方法.分享给大家供大家参考,具体如下: 多维数组其实就是多个一维数组的嵌套,Python中有原生的list,类似一个动态数组. 所以动态生成多维数组的思想就是在list中动态嵌套添加list. 下面代码生成一个一个3×3×2的三维数组: # coding:utf-8 # 使用Python3中的print函数 from __future__ import print_function arr = [] # 基本思想是在list中动态添加list,每个li

  • python生成多个只含0,1元素的随机数组或列表的实例

    如下所示: >>> import numpy as np >>> myarray= np.random.randint(0,2,10)输出只含0,1元素的一维数组,长度为10 >>> myarray array([1, 1, 1, 0, 1, 0, 0, 1, 1, 0]) >>> myarray= np.random.randint(0,2,(3,10))输出只含0,1元素的3行10列数组 >>> myarray

  • python由已知数组快速生成新数组的方法

    需求描述 在利用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:已知数

  • Python使用dict.fromkeys()快速生成一个字典示例

    本文实例讲述了Python使用dict.fromkeys()快速生成一个字典.分享给大家供大家参考,具体如下: >>> result = dict.fromkeys(range(24), 0) >>> result {0: 0, 1: 0, 2: 0, 3: 0, 4: 0, 5: 0, 6: 0, 7: 0, 8: 0, 9: 0, 10: 0, 11: 0, 12: 0, 13: 0, 14: 0, 15: 0, 16: 0, 17: 0, 18: 0, 19: 0

  • 对Python中9种生成新对象的方法总结

    先定义一个类: class Point: def __init__(self, x, y): self.x = x self.y = y 下面我们使用9种方法来生成新的对象: point1 = Point(1, 2) point2 = eval("{}({}, {})".format("Point", 1, 2)) point3 = globals()["Point"](1, 2) point4 = locals()["Point&qu

  • python实现将元祖转换成数组的方法

    本文实例讲述了python实现将元祖转换成数组的方法.分享给大家供大家参考.具体分析如下: python的元祖使用一对小括号表示的,元素是固定的,如果希望添加新的元素,可以先将元祖转换成数组列表,再进行操作 colour_tuple = ("Red","Green","Blue") colour_list = list(colour_tuple) assert colour_list == ["Red","Green

  • idea快速生成代码配置的方法示例

    idea快速生成代码配置的方法示例

    前言 这里是用的goland idea,实际上这个idea和 intellij idea的配置是一样的,并没有太大区别, 开整 1.进入 File->settings->Editor->Live Templates 2.添加模板组 点击右上角的绿色+号,然后选择template group ,然后输入group的name,这个名字你随便起,我的是:my.然后点ok 属于你自己的组就创建好了 3. 添加模板 选中刚才创建的myGroup,然后再次点击右侧的绿色+,这次选择的是第一个1. L

  • Python实现抓取网页生成Excel文件的方法示例

    本文实例讲述了Python实现抓取网页生成Excel文件的方法.分享给大家供大家参考,具体如下: Python抓网页,主要用到了PyQuery,这个跟jQuery用法一样,超级给力 示例代码如下: #-*- encoding:utf-8 -*- import sys import locale import string import traceback import datetime import urllib2 from pyquery import PyQuery as pq # 确定运行

  • C#实现保存文件时重名自动生成新文件的方法

    本文实例讲述了C#实现保存文件时重名自动生成新文件的方法.分享给大家供大家参考.具体如下: 将一个文档保存为 a.txt 时,发现此文件已经存在,则自动保存为 a(1).txt /// <summary> /// Generates a new path for duplicate filenames. /// </summary> /// <param name="path">The path.</param> /// <retu

  • php实现根据字符串生成对应数组的方法

    本文实例讲述了php实现根据字符串生成对应数组的方法,是比较实用的技巧.分享给大家供大家参考.具体方法如下: 先看看如下示例: <?php $config = array( 'project|page|index' => 'content', 'project|page|nav' => array( array( 'image' => '1.jpg', 'name' => 'home' ), array( 'image' => '2.jpg', 'name' =>

  • pandas通过loc生成新的列方法

    pandas中一个很便捷的使用方法通过loc.iloc.ix等索引方式,这里记录一下: df.loc[条件,新增列] = 赋初始值 如果新增列名为已有列名,则在原来的数据列上改变 import pandas as pd import numpy as np data = pd.DataFrame(np.random.randint(0,100,40).reshape(10,4),columns=list('abcd')) print(data) data.loc[data.d >= 50,'大于

随机推荐