python - リストから集計範囲を生成する

Question

整数のリストがあります：

list = [a,b,c,d]

これらの範囲に変換したいと思います：

0..a
a+1..a+b
a+b+1..a+b+c
a+b+c+1..a+b+c+d

Pythonでこれを行う方法はありますか?

ありがとう

Answer 1

受け入れられた答えは正しい結果をもたらしますが、いくつかのループとリストジェネレーターを使用するだけで解決策を得ることができます。質問がそのような単純な問題について尋ねているので、より高度な言語演算子は不適切に思えます - いくつかのリストを生成します。

このソリューションは、正の整数に対してのみ機能します。ネガを含むシーケンスの処理は、読者の演習として残されています。

# starting data
nums = [3, 5, 22, 6]

# Generate endpoints for output sequences.
# Note that the first sequence starts specially at 0.
endpoints = [0]
for i in range(len(nums)):
    endpoints.append(endpoints[i] + nums[i])
endpoints[0] = -1

# Generate output sequences.
outseqs = []
for i in range(1, len(endpoints)):
    outseq = [n for n in range(endpoints[i - 1] + 1, endpoints[i] + 1)]
    outseqs.append(outseq)

# Display output sequences.
for i in range(len(outseqs)):
    print outseqs[i]

Answer 2

これまでに提供されたすべてのソリューションは、入力に負の数値が含まれている場合は機能しないため、逆順の範囲が必要になります。私の解決策はそれをカバーしています。Python 2 および 3 で動作します。

from itertools import izip

# create the x (=start) and y (=stop) coordinates for the ranges separately
def _get_xs(iterable):
    yield 0
    for i in xrange(1, len(iterable)):
        yield sum(iterable[:i]) + 1

def _get_ys(iterable):
    yield iterable[0]
    for i in xrange(1, len(iterable)):
        yield sum(iterable[:i+1])

def to_ranges(iterable):
    xs = _get_xs(iterable)
    ys = _get_ys(iterable)
    for x, y in izip(xs, ys):
        if x < y:
            step = 1
            y += 1
        elif x > y:
            step = -1
            y -= 1
        else:
            step = 0
        try:
            yield range(x, y, step)
        except ValueError:
            yield [x]

例:

# edge case: instead of malformed ranges such as range(10, 10), put [10] instead

>>> list(to_ranges([1, 2, 3, 4]))
[[0, 1], [2, 3], [4, 5, 6], [7, 8, 9, 10]]

>>> list(to_ranges([4, 3, 2, 1]))
[[0, 1, 2, 3, 4], [5, 6, 7], [8, 9], [10]]

>>> list(to_ranges([4, 3, 2, -1]))
[[0, 1, 2, 3, 4], [5, 6, 7], [8, 9], [10, 9, 8]]

>>> list(to_ranges([-4, -3, -2, -1]))
[[0, -1, -2, -3, -4], [-3, -4, -5, -6, -7], [-6, -7, -8, -9], [-8, -9, -10]]

>>> list(to_ranges([-1, -2, -3, -4]))
[[0, -1], [0, -1, -2, -3], [-2, -3, -4, -5, -6], [-5, -6, -7, -8, -9, -10]]