按可能性组织数据并使用 Python 中新组织的值创建新列

Question

我对 Python 完全陌生，所以如果我的问题不完全清楚或太明显，我将感谢您的耐心等待。

我需要使用条件组织数据，然后使用新组织的数据创建新列。我创建了一个简短的示例来说明我的观点。假设 S 是我要组织和拆分的数据。

这是我需要做的：

1. 通过计算相邻元素的比率来查找它们的可能性。例如，如果1.3 > S[i]/S[i+1] > 0.80，值应该存储在一个系列中，比如a。

2. 如果不满足条件，则创建一个新系列b 以填充下一个S[i+1]，直到满足条件。

3. 否则，创建一个新系列c...并重新开始。

示例

S=pd.Series([9,9,10,11,12,10,22,23,21,20,21,33,35,37,34,50,51,56,48,51])

a =[]
b = []
c = []
d = []

i=0
j = 1
k=1
m=1


for i in range(0,19,1):
    if 1.3 > S[i]/S[i+1] > 0.80: 
        a.append(S[i])
        j +=1
    else:
      break

for i in range(j,19,1):
      if 1.3 > S[i]/S[i+1] > 0.80: 
          b.append(S[i])
          k +=1
      else:
          break

for i in range(j+k,19,1):
    if 1.3 > S[i]/S[i+1] > 0.80: 
        c.append(S[i])
        m +=1
    else:
        break


for i in range(j+k+m,19,1):
    if 1.3 > S[i]/S[i+1] > 0.80:
        d.append(S[i])
    else:
        break


print('a=',a,'b=',b,'c=',c,'d=',d)

a= [9, 9, 10, 11, 12] 
b= [22, 23, 21, 20] 
c= [33, 35, 37] 
d= [50, 51, 56, 48]

这段代码有两个主要问题：

首先，有一个问题。缺少每个生成的系列的最后一个数字。例如，在系列b 中，缺少术语21。

其次，是否可以根据需要使用循环来生成系列？真实数据有 10000+ 行，可能需要数百个不同的系列。

感谢您的帮助！

Answer 1

我想我明白你在问什么，但不是 100% 肯定。假设我已经正确解释了这个问题，这就是我认为您正在寻找的内容。

def bucket_nums(numbers):
    groups = collections.defaultdict(list)
    i = 0
    while i < len(numbers) - 1:
        j = i
        #bug here: num[i+1] gets incremented before the bounds test
        while 1.3 > numbers[i]/numbers[i+1] > 0.8 and i < len(numbers)- 2: #also should be -1, not -2 (this is why it skipped the last one)
            groups['group{}'.format(j)].append(numbers[i])
            i += 1
        groups['group{}'.format(j)].append(numbers[i])
        i += 1
    return groups

编辑：

你是对的，代码有错误。在可能增加超过列表长度之后，我正在测试索引边界，然后通过将索引检查减少到< [i+2] 来错误地补偿它。下面是固定代码。离开未修复的版本公开羞辱自己懒惰并试图以太快的速度给出答案。

def bucket_nums(numbers):
    groups = collections.defaultdict(list)
    i = 0
    while i < len(numbers):
        j = i
        while i < len(nums) - 1:
            if  1.3 > numbers[i] / numbers[i+1] > 0.8:  
                groups['group{}'.format(j)].append(numbers[i])
                i += 1
            else:
                break
        groups['group{}'.format(j)].append(numbers[i])
        i += 1
    return groups