Advertisement

Python与IDL where函数对比分析

阅读量:

Python where()函数与IDL where()函数的比较

近期在使用Python的过程中,我对其中的where函数产生了一些困惑,因其与IDL语言中的对应功能存在差异。本文将对这两种编程语言中where函数的相关特性展开简要对比分析。

1 . Python 的where函数在多条件下的判断

复制代码
    import numpy as np
    
    #首先创建两个非常大的随机数组用来做测试
    A = np.random.randn(10000, 10000)
    B = np.random.randn(10000, 10000) 
    
    index = np.where( (A > -1) & (A < 1) & (B > 10) & (B < 20) )
    print(type(index), np.shape(index))
    
    
    AI写代码python
    
    运行

观察np.where函数返回的索引值

复制代码
    #得到满足条件的下标index是一个两行N列的元组,第一行为“行下标”,第二行为“列下标”
    <class 'tuple'> (2, 9280675)
    
    
    AI写代码python
    
    运行

全部评论 (0)

还没有任何评论哟~