欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

numpy ndarray 按條件篩選數(shù)組,關(guān)聯(lián)篩選的例子

 更新時間：2019年11月26日 09:32:42   作者：y小川  
今天小編就為大家分享一篇numpy ndarray 按條件篩選數(shù)組,關(guān)聯(lián)篩選的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

最近的項目中大量涉及數(shù)據(jù)的預(yù)處理工作，對于ndarray的使用非常頻繁。其中ndarray如何進(jìn)行數(shù)值篩選，總結(jié)了幾種方法。

1.按某些固定值篩選

如下面這段代碼從，ndarray中可以篩選出數(shù)值等于3的子數(shù)組和其在原數(shù)組中的索引位置。

import numpy as np

arr = np.array([1, 1, 1, 134, 45, 3, 46, 45, 65, 3, 23424, 234, 12, 12, 3, 546, 1, 2])
print(np.where(arr==3))
print(arr[np.where(arr == 3)])

Output:

(array([ 5, 9, 14], dtype=int32),)
[3 3 3]

2.按多個固定值篩選

按上述方法篩選多個固定值也是可行的，將不同條件用括號括起來，之間打 | 即可。

import numpy as np

arr = np.array([1, 1, 1, 134, 45, 3, 46, 45, 65, 3, 23424, 234, 12, 12, 3, 546, 1, 2])
print(np.where((arr == 3) | (arr == 1)))
print(arr[np.where((arr == 3) | (arr == 1))])

Output:

(array([ 0, 1, 2, 5, 9, 14, 16], dtype=int32),)
[1 1 1 3 3 3 1]

3.按范圍篩選

除了按固定值，我們還可以按一定的范圍進(jìn)行篩選

import numpy as np

arr = np.array([1, 1, 1, 134, 45, 3, 46, 45, 65, 3, 23424, 234, 12, 12, 3, 546, 1, 2])
print(np.where(arr > 3))
print(arr[np.where(arr > 3)])

Output:

(array([ 3, 4, 6, 7, 8, 10, 11, 12, 13, 15], dtype=int32),)
[ 134  45  46  45  65 23424  234  12  12  546]

如果不需要index，還可以有更快的方法

import numpy as np

arr = np.array([1, 1, 1, 134, 45, 3, 46, 45, 65, 3, 23424, 234, 12, 12, 3, 546, 1, 2])
print(arr[arr > 3])

Output:

[ 134  45  46  45  65 23424  234  12  12  546]

那為什么還需要用np.where呢？因為索引可以滿足不同數(shù)組間的篩選，比如有a, b 兩個數(shù)組我們需要篩選出所有a == 0的b時即可使用np.where函數(shù)。

import numpy as np

a = np.array([0, 0, 1, 1, 0, 1, 1, 0, 1])
b = np.arange(len(a))
print(a)
print(b)
print(b[np.where(a == 0)])

Output:

[0 0 1 1 0 1 1 0 1]
[0 1 2 3 4 5 6 7 8]
[0 1 4 7]

以上這篇numpy ndarray 按條件篩選數(shù)組,關(guān)聯(lián)篩選的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

相關(guān)文章

最新評論