1

我有一个如下列表， 

list_ = [('a', 5), ('a', 6), ('a', 8), ('a', 2), ('a', 5), ('a', 9), ('a', 10),
             ('b', 3), ('b', 7), ('b', 6), ('b', 2), ('b', 6), ('b', 5), ('b', 10)]

我想做以下操作。如果元组中的值增加，我必须跟踪增量，如果它减少，我必须跟踪减量，最后我需要两者的总和。

# for key 'a' for example
# increment      decrement
#       1             6
#       2
#       3
#       4
#       1
# Hence total increment is 11 and decrement is 6
# so I need the result like this. ('a', (11, 6))

例如，我确实喜欢这样做以获得增量，但没有工作。

from collections import defaultdict

d = defaultdict(list)
for k, v in list_:
    d[k].append(v)

counter1 = 0

for key, value in d.items():
  for i in range(len(value)):
    try:
      if value[i] < value[i+1]:
        s = value[i+1] - value[i]
        counter1 +=s
    except:
      pass


print(counter1)
4

2 回答 2

1 
df = pd.DataFrame([('a', 5), ('a', 6), ('a', 8), ('a', 2), ('a', 5), ('a', 9), ('a', 10),
             ('b', 3), ('b', 7), ('b', 6), ('b', 2), ('b', 6), ('b', 5), ('b', 10)], columns=['col1','col2'])


for value,tmp_df  in df.groupby('col1'):

    increase = tmp_df['col2']-tmp_df['col2'].shift(1)
    positive = increase[increase>0].sum()
    negative = increase[increase<0].sum()
    print ((value, (positive, negative)))
于 2019-11-17T13:00:08.053 回答
1 
def incdec(lst):
    d = {}
    for k, v in lst:
        try:
            last = d[k]
        except KeyError:
            d[k] = [0, 0, v]
        else:
            delta = v - last[2]
            last[2] = v
            last[delta < 0] += abs(delta)
    return [(k, (v[0], v[1])) for k, v in d.items()]

它适用于可迭代lst（不收集数据，只记住增量和减量，而不是整个数据）。

d是临时列表的字典，其中 [0] 是增量，[1] 是减量 [2] 最后一个值。

last[delta < 0]是一个技巧，因为bool是一种int，那么在这种情况下，您可以寻址 [0] (inc) 或 [1] (dec)。我知道，它不太可读。您可以将其更改为if / else.

最后一行以所需格式返回数据（inc 和 dec，不带 [2] – 最后一个值）。

于 2019-11-17T13:02:08.460 回答