我在分割数据文件中的文本时遇到问题,假设数据文件包含:
Row 1
apple
bob
cat
dog
ear
fun
Row 2
glow
horse
idea
joke
kick
lemon
Row 3
money
new
odd
park
queen
run
我想拆分它,使它成为一个嵌套列表,如下所示:
[[apple, bob], [cat, dog], [ear, fun]],
[[glow, horse], [idea, joke], [kick, lemon]],
[[money, new], [odd, park], [queen, run]]
到目前为止,这是我的工作:
def text_file(data_file):
nested_list = []
main_list = []
my_list = ''
for index in data_file:
index = index.strip()
if (index in my_list):
main_list.append(nested_list)
nested_list = []
else:
nested_list.append(index)
if (nested_list):
main_list.append(nested_list)
return (main_list)
但这会返回:
text_file(open("data_file.txt", "r"))
[['Row 1', 'apple', 'bob', 'cat', 'dog', 'ear', 'fun'],
['Row 2', 'glow', 'horse', 'idea', 'joke', 'kick', 'lemon'],
['Row 3', 'money', 'new', 'odd', 'park', 'queen', 'run']]
如果没有导入任何内容,我该如何实现?如果可能的话,我可以在代码中添加什么?
答案 0 :(得分:5)
您需要做的是将文件分割为\n\n
(两个换行符),这将为您提供组,然后逐行拆分结果,然后使用zip
跳过文件适当地建立你需要的清单,例如:
s = """Row 1
apple
bob
cat
dog
ear
fun
Row 2
glow
horse
idea
joke
kick
lemon
Row 3
money
new
odd
park
queen
run"""
lines = s.split('\n\n')
for line in lines:
words = line.splitlines()
print([ [i, j] for i, j in zip(words[1::2], words[2::2]) ])
[['apple', 'bob'], ['cat', 'dog'], ['ear', 'fun']]
[['glow', 'horse'], ['idea', 'joke'], ['kick', 'lemon']]
[['money', 'new'], ['odd', 'park'], ['queen', 'run']]
答案 1 :(得分:1)
类似的内容,使用regex
和iterators
。
在regex
处使用Row number
拆分,然后您可以使用zip
或iterator
来获得预期的输出。
In [8]: with open("data.txt") as f:
spl=re.split(r"Row \d+",f.read())[1:]
for x in spl:
sp=x.split()
it=iter(sp)
print ([[next(it),next(it)] for _ in range(len(sp)//2)])
...:
[['apple', 'bob'], ['cat', 'dog'], ['ear', 'fun']]
[['glow', 'horse'], ['idea', 'joke'], ['kick', 'lemon']]
[['money', 'new'], ['odd', 'park'], ['queen', 'run']]
答案 2 :(得分:0)
if (nested_list):
new_list = nested_list[1:]
main_list.append(zip(new_list[::2], new_list[1::2]))
试试这个
上面的代码不是在主列表中附加嵌套列表,而是首先形成连续元素对,然后将其附加。