脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Python np.setdiff1d()函数

Python中的np.setdiff1d()函数详解

 更新时间：2024年06月18日 15:17:04   作者：对许  
Python中的np.setdiff1d()函数可用于找出两个序列集合中元素的差异,下面通过示例代码给大家详细讲解,感兴趣的朋友跟随小编一起看看吧

Python中的np.setdiff1d()函数可用于找出两个序列集合中元素的差异

API及参数如下：

np.setdiff1d(ar1,ar2,assume_unique)：找出序列1在序列2中的差异，并返回序列1中不在序列2中的元素

  • ar1：输入数组
  • ar2：输入比较数组
  • assume_unique：是否假设唯一，默认为False，输出去重且元素从小到大排序，如果为True，则不会去重，输出顺序保持原来的顺序

以下是一个使用示例：

import numpy as np
a = np.array([3, 2, 2, 1])
b = np.array([5, 4, 3, 3])
# a = [3, 2, 2, 1]
# b = [5, 4, 3, 3]
c = np.setdiff1d(a, b)
print(c)    # [1 2]
c = np.setdiff1d(a, b, assume_unique=True)
print(c)    # [2 2 1]

补充：python numpy中setdiff1d的用法

一、函数解释

setdiff1d(ar1, ar2, assume_unique=False)

1.功能：找到2个数组中集合元素的差异。

2.返回值：在ar1中但不在ar2中的已排序的唯一值。

3.参数：

  • ar1：array_like 输入数组。
  • ar2：array_like 输入比较数组。
  • assume_unique：bool。如果为True，则假定输入数组是唯一的，即可以加快计算速度。 默认值为False。

二、具体示例

1.assume_unique = False的情况： 

    a = np.array([1,2,3])
    b = np.array([4,5,6])
    c = np.setdiff1d(a, b)
    print(c)#[1 2 3]
    a = np.array([1,2,3])
    b = np.array([1,2,3])
    c = np.setdiff1d(a, b)
    print(c)#[]
    a = np.array([1,2,3])
    b = np.array([2,3,4])
    c = np.setdiff1d(a, b)
    print(c)#[1]
    a = np.array([1,2,3,4])
    b = np.array([3,4,5,6])
    c = np.setdiff1d(a, b)
    print(c)#[1 2]
    a = np.array([1,2,3,2,4,1])
    b = np.array([3,4,5,6])
    c = np.setdiff1d(a, b)
    print(c)#[1 2]
    a = np.array([8,2,3,2,4,1])
    b = np.array([7,4,5,6,3])
    c = np.setdiff1d(a, b)
    print(c)#[1 2 8]

可以从最后看出返回的值从小到大排序，并且唯一。（8在a的第1位，2在a中重复了2次）

2.assume_unique = True的情况： ​​​​​​​

    a = np.array([3,2,1])
    b = np.array([4,5,6])
    c = np.setdiff1d(a, b,True)
    print(c)#[3 2 1]
    a = np.array([8,2,3,2,4,1])
    b = np.array([7,4,5,6,3])
    c = np.setdiff1d(a, b,True)
    print(c)#[8 2 2 1]
    a = np.array([8,2,3,4,2,4,1])
    b = np.array([7,9,5,6,3])
    c = np.setdiff1d(a, b,True)
    print(c)#[8 2 4 2 4 1]

可以看出把在a中的但是不在b中的元素按a中的顺序排序，并且不合并重复的元素，即假定输入数组也是唯一的，因此相比于False确实提升了运算速度。

三、整体代码

import numpy as np
def main():
    a = np.array([1,2,3])
    b = np.array([4,5,6])
    c = np.setdiff1d(a, b)
    print(c)#[1 2 3]
    a = np.array([1,2,3])
    b = np.array([1,2,3])
    c = np.setdiff1d(a, b)
    print(c)#[]
    a = np.array([1,2,3])
    b = np.array([2,3,4])
    c = np.setdiff1d(a, b)
    print(c)#[1]
    a = np.array([1,2,3,4])
    b = np.array([3,4,5,6])
    c = np.setdiff1d(a, b)
    print(c)#[1 2]
    a = np.array([1,2,3,2,4,1])
    b = np.array([3,4,5,6])
    c = np.setdiff1d(a, b)
    print(c)#[1 2]
    a = np.array([8,2,3,2,4,1])
    b = np.array([7,4,5,6,3])
    c = np.setdiff1d(a, b)
    print(c)#[1 2 8]
    a = np.array([3,2,1])
    b = np.array([4,5,6])
    c = np.setdiff1d(a, b,True)
    print(c)#[3 2 1]
    a = np.array([8,2,3,2,4,1])
    b = np.array([7,4,5,6,3])
    c = np.setdiff1d(a, b,True)
    print(c)#[8 2 2 1]
    a = np.array([8,2,3,4,2,4,1])
    b = np.array([7,9,5,6,3])
    c = np.setdiff1d(a, b,True)
    print(c)#[8 2 4 2 4 1]
if __name__ == '__main__':
    main()

到此这篇关于Python中的np.setdiff1d()函数的文章就介绍到这了,更多相关Python np.setdiff1d()函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

相关文章

  • python使用rsa加密算法模块模拟新浪微博登录

    python使用rsa加密算法模块模拟新浪微博登录

    这篇文章主要介绍了python使用rsa加密算法模块模拟新浪微博登录的示例，大家参考使用吧
    2014-01-01
  • python实现双向链表原理

    python实现双向链表原理

    这篇文章主要为大家详细介绍了python实现双向链表原理，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2022-05-05
  • Python常见异常类型处理

    Python常见异常类型处理

    这篇文章主要个大家分享的是Python常见异常类型处理，编写程序的过程，难免会遇到一些错误，程序逻辑又或者运行与系统发生各种冲突等等，下面我们就来看看那些我们在编程中常遇到的异常都是怎么处理的吧，需要的小伙伴可以参考一下
    2021-12-12
  • 利用Python pyecharts绘制饼图

    利用Python pyecharts绘制饼图

    这篇文章主要介绍了利用Python pyecharts绘制饼图，饼图主要用于表现不同类目的数据在总和中的占比。每个的弧度不是数据量的占比，下面我们一起进入文章看看具体的实现过程吧,需要的朋友也可以参考一下
    2021-12-12
  • Python调试神器之PySnooper的使用教程分享

    Python调试神器之PySnooper的使用教程分享

    对于每个程序开发者来说，调试几乎是必备技能。本文小编就来给大家介绍一款非常好用的调试工具，它能在一些场景下，大幅度提高调试的效率， 那就是 PySnooper，希望大家喜欢
    2023-02-02
  • Python操作Excel的学习笔记

    Python操作Excel的学习笔记

    这篇文章主要介绍了Python操作Excel的学习笔记，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-02-02
  • 解决python3爬虫无法显示中文的问题

    解决python3爬虫无法显示中文的问题

    下面小编就为大家分享一篇解决python3爬虫无法显示中文的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-04-04
  • Python实现线性搜索算法的示例代码

    Python实现线性搜索算法的示例代码

    线性搜索算法,也称为顺序搜索算法,是一种简单但常用的搜索技术,在本文中,将深入研究线性搜索算法,并演示如何在 Python 中实现它,需要的可以参考下
    2024-02-02
  • Python下的subprocess模块的入门指引

    Python下的subprocess模块的入门指引

    这篇文章主要介绍了Python下的subprocess模块的入门指引,subprocess模块被用于Python的多线程编程,需要的朋友可以参考下
    2015-04-04
  • Django用户登录与注册系统的实现示例

    Django用户登录与注册系统的实现示例

    这篇文章主要介绍了Django用户登录与注册系统的实现示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-06-06

最新评论