0

次のような構造でいくつかのデータをビニングしたとします。

data = {(1,1): [...] # list of float,
        (1,2): [...],
        (1,3): [...],
        (2,1): [...],
        ... }

ここでは、ビニング用の軸は2つしかありませんが、N個あるとします。たとえば、N = 3の軸があり、2番目のビンが1のデータが必要な場合、関数が必要です。

(None, 1, None) -> [(1, 1, 1), (1, 1, 2), (1, 1, 3), ...
                    (2, 1, 1), (2, 1, 2), (2, 1, 3), ...]

itertools.chain結果に使用できるように

あなたはからすべての軸の範囲を知っています：

axes_ranges = [(1, 10), (1, 8), (1, 3)]

その他の例：

(None, 1, 2) -> [(1, 1, 2), (2, 1, 2), (3, 1, 2), ...]
(None, None, None) -> all the combinations
(1,2,3) -> [(1,2,3)]
4

3 に答える 3

1

うーん、どうですか：

import itertools

def combinations_with_fixpoint(iterables, *args):
    return itertools.product(*([x] if x else y for x, y in zip(args, iterables)))


axes_ranges = [(1, 7), (1, 8), (77, 79)]

combs = combinations_with_fixpoint(
    itertools.starmap(range, axes_ranges),
    None, 5, None
)

for p in combs:
    print p

# (1, 5, 77)
# (1, 5, 78)
# (2, 5, 77)
# (2, 5, 78)
# (3, 5, 77)
# (3, 5, 78)
# (4, 5, 77)
# (4, 5, 78)
# (5, 5, 77)
# (5, 5, 78)
# (6, 5, 77)
# (6, 5, 78)

複数の「フィックスポイント」を許可するためにリストを渡すだけかもしれません。

def combinations_with_fixpoint(iterables, *args):
    return itertools.product(*(x or y for x, y in zip(args, iterables)))

combs = combinations_with_fixpoint(
    itertools.starmap(range, axes_ranges),
    None, [5, 6], None
)
于 2012-05-15T09:58:34.847 に答える
1

あなたが車輪の再発明をしているように思えます。おそらく使用したいのはnumpy.ndarrayです。

    import numpy as np
    >>> x = np.arange(0,27)
    >>> x
    array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
    17, 18, 19, 20, 21, 22, 23, 24, 25, 26])
    >>> x.reshape(3,3,3)
    array([[[ 0,  1,  2],
    [ 3,  4,  5],
    [ 6,  7,  8]],

    [[ 9, 10, 11],
     [12, 13, 14],
     [15, 16, 17]],

    [[18, 19, 20],
     [21, 22, 23],
     [24, 25, 26]]])

    >>> x[0]
    array([[0, 1, 2],
    [3, 4, 5],
    [6, 7, 8]])
    >>> x[:,1,:]
    array([[ 3,  4,  5],
    [12, 13, 14],
    [21, 22, 23]])
    >>> x[:,1,1]
    array([ 4, 13, 22])

これはN次元を持つことができます。インデックス付けが3次元の例では、x [a、b、c] = x [layer、row、column]の立方体として表示できます。インデックスとして「：」を使用することは、単に「すべて」を意味します

于 2012-05-15T10:01:37.640 に答える
0 
binning = [[0, 0.1, 0.2], [0, 10, 20], [-1, -2, -3]]
range_binning = [(1, len(x) + 1) for x in binning]

def expand_bin(thebin):
    def expand_bin_index(thebin, freeindex, rangebin):
        """
        thebin = [1, None, 3]
        freeindex = 1
        rangebin = [4,5]
        -> [[1, 4, 3], [1, 5, 3]]
        """
        result = []
        for r in rangebin:
            newbin = thebin[:]
            newbin[freeindex] = r
            result.append(newbin)
        return result

    tmp = [thebin]
    indexes_free = [i for i,aa in enumerate(thebin) if aa is None]
    for index_free in indexes_free:
        range_index = range(*(range_binning[index_free]))
        new_tmp = []
        for t in tmp:
            for expanded in expand_bin_index(t, index_free, range_index):
                new_tmp.append(expanded)
        tmp = new_tmp
    return tmp

inputs = ([None, 1, 2], [None, None, 3], [None, 1, None], [3, 2, 1], [None, None, None])
for i in inputs:
    print "%s-> %s" % (i, expand_bin(i))
于 2012-05-15T10:01:03.730 に答える