检查Python列表中的下一个对象

Question

我有一个包含以下信息结构的list （来自CSV）：

Item 1
NUMBER Random ID
Item 2
NUMBER Random ID
Item 3
Item 4
Item 5
NUMBER Random ID

我想创建一个新的list （CSV），如下所示：

Item 1 NUMBER Random ID
Item 2 NUMBER Random ID
Item 5 NUMBER Random ID

因此，如果下一行不包含字符串NUMBER ，那么我想从Item 1,2,3...及其下一行创建一个字符串。

我可以读取CSV并将其用作列表，但是我不知道如何跟踪行。 我的第一个想法是创建一个字典列表，其中每个字典都包含该行及其内容的索引号，然后我可以遍历list_with_dicts并检查原始列表的下一行。

raw_list = [] 
num = 0
list_with_dicts = []
for x in raw_list:

    num2 = num + 1
    dict1 = {}
    dict1['index'] = num2
    dict1['çontent'] = x 
    list_with_dicts.append(dict1)

for d in list_with_dicts:

    number_of_next_line = d['index'] + 1    

    if "NUMBER" in raw_list[number_of_next_line]:
        new_string = "%s %s" % (d[content], raw_list[number_of_next_line])
    else:
        print("String without number")

但是我不确定这是最简单，最好的方法，因此如果有人可以向我展示一个更简单的解决方法，我将不胜感激。

Answer 1

好玩的问题！

raw_list = ["Item 1",
            "NUMBER Random ID1",
            "Item 2",
            "NUMBER Random ID2",
            "Item 3",
            "Item 4",
            "Item 5",
            "NUMBER Random ID5"]

clean_list = [raw_list[i]+" "+raw_list[i+1] for i in range(0,len(raw_list),2) if "Item" not in raw_list[i+1]]
print clean_list

输出：

['Item 1 NUMBER Random ID1', 'Item 2 NUMBER Random ID2', 'Item 5 NUMBER Random ID5']

您还可以使用zip使其更短，但可读性可能更低：

clean_list1 = [i1+" "+i2 for i1,i2 in zip(raw_list[::2],raw_list[1::2]) if "Item" not in i2]
print clean_list1

Answer 2

这个问题的处理方法略有不同-搜索包含字符串NUMBER ，然后将该行与上一行连接。 这将产生更简单的代码：

l = ['Item 1', 'NUMBER Random ID', 'Item 2', 'NUMBER Random ID', 'Item 3', 'Item 4', 'Item 5', 'NUMBER Random ID']

result = []
for i, s in enumerate(l[1:], 1):
    if 'NUMBER' in s:
       result.append('{} {}'.format(l[i-1], s))

或作为列表理解：

result = ['{} {}'.format(l[i-1], s) for i,s in enumerate(l[1:], 1) if 'NUMBER' in s]

目前尚不清楚期望输出什么-您提到CSV，这表示输出列表应包含单个字段，在这种情况下，结果应为列表列表。 像这样：

result = [[l[i-1], s] for i,s in enumerate(l[1:], 1) if 'NUMBER' in s]

这将创建以下列表列表：

[['Item 1', 'NUMBER Random ID'],
 ['Item 2', 'NUMBER Random ID'],
 ['Item 5', 'NUMBER Random ID']]

可以使用csv模块轻松将其保存到CSV文件中：

import csv

with open('result.csv', 'w') as f:
    csv.writer(f).writerows(result)

Answer 3

使用enumerate(<list>)可以迭代索引和元素，因此可以轻松检查下一个元素：

result = []
for i, val in enumerate(lst):
    if i == len(lst) - 1:
        break # to avoid IndexError
    if lst[i + 1][:3] == 'NUM':
        result.append('%s %s' % (val, lst[i + 1])

功能编程版本：

result = \
    list(
        map(
            lambda i: 
                 '%s %s' % (lst[i - 1], lst[i]),
            filter(
                lambda i:
                    lst[i][:3] == 'NUM',
                range(1, len(lst))
            )
        )
    )

Answer 4

使用列表理解：

result = ["%s %s" % (x,raw_list[i+1]) for i, x in enumerate(raw_list) 
                if i < len(raw_list)-1 and 'NUMBER' in raw_list[i+1]]

Answer 5

new_list=[]
i=0
while i < len(raw_list)-1:
    if raw_list[i+1][:len("NUMBER")] == "NUMBER":
        new_list.append("%s %s" % (raw_list[i], raw_list[i+1]))
        i=i+2
    else:
        i=i+1

Answer 6

result = []
i, l = 0, len(raw_input)
while i < l:
  if 'item' in raw_input[i]:
    result.append(raw_input[i])
  else:
    result[-1] += raw_input[i]
  i += 1
return filter(lambda x: 'random' in x.lower(), result)