在numpy数组中对类似值进行分组

Question

我正在尝试在我的numpy数组中对非常相似的值进行分组。 46.29821447,49.49781571,49.83072758,50.89081787,98.49113721,98.5522082,99.29547499,99.91765345,99.93779431,99.95351796,99.98066963,99.99294867,100

1. 查看数组中的值
2. 获取与49.x 98.x 99.x
类似的值
3. 将他们分组
4. 获得每组的最小值和最大值。

Answer 1

搜索“Python group by”会引导您使用groupby方法获取itertools的文档。在Stack Overflow上有关于如何使用它的examples。

from itertools import groupby

l = [46.29821447,49.49781571,49.83072758,50.89081787,98.49113721,98.5522082,99.29547499,99.91765345,99.93779431,99.95351796,99.98066963,99.99294867,100]

for key, group in groupby(l, round):
    s = sorted(group)
    print key, s[0], s[-1]

输出：

46.0 46.29821447 46.29821447
49.0 49.49781571 49.49781571
50.0 49.83072758 49.83072758
51.0 50.89081787 50.89081787
98.0 98.49113721 98.49113721
99.0 98.5522082 99.29547499
100.0 99.91765345 100

---

<强>更新

使用NumPy的floor方法......

from itertools import groupby
import numpy as np

l = [46.29821447,49.49781571,49.83072758,50.89081787,98.49113721,98.5522082,99.29547499,99.91765345,99.93779431,99.95351796,99.98066963,99.99294867,100]

for key, group in groupby(l, np.floor):
    s = sorted(group)
    print key, s[0], s[-1]

输出：

46.0 46.29821447 46.29821447
49.0 49.49781571 49.83072758
50.0 50.89081787 50.89081787
98.0 98.49113721 98.5522082
99.0 99.29547499 99.99294867
100.0 100 100