问题描述
我有一个文件，我想从中创建一个列表（“timestep”），其中的数字出现在每行“ITEM：timestep”之后，所以：
timestep = [253400,253500,.. etc]
这是我拥有的文件示例：
ITEM: timestep
253400
ITEM: NUMBER OF ATOMS
378
ITEM: Box BOUNDS pp pp pp
-2.6943709180241954e-01 5.6240920636804063e+01
-2.8194230631882372e-01 5.8851195163321044e+01
-2.7398090193568775e-01 5.7189372326936599e+01
ITEM: ATOMS id type q x y z
16865 3 0 28.8028 1.81293 26.876
16866 2 0 27.6753 2.22199 27.8362
16867 2 0 26.8715 1.04115 28.4178
16868 2 0 25.7503 1.42602 29.4002
16869 2 0 24.8716 0.25569 29.8897
16870 3 0 23.7129 0.593415 30.8357
16871 3 0 11.9253 -0.270359 31.7252
ITEM: timestep
253500
ITEM: NUMBER OF ATOMS
378
ITEM: Box BOUNDS pp pp pp
-2.6943709180241954e-01 5.6240920636804063e+01
-2.8194230631882372e-01 5.8851195163321044e+01
-2.7398090193568775e-01 5.7189372326936599e+01
ITEM: ATOMS id type q x y z
16865 3 0 28.8028 1.81293 26.876
16866 2 0 27.6753 2.22199 27.8362
16867 2 0 26.8715 1.04115 28.4178
16868 2 0 25.7503 1.42602 29.4002
16869 2 0 24.8716 0.25569 29.8897
16870 3 0 23.7129 0.593415 30.8357
16871 3 0 11.9253 -0.270359 31.7252
为此，我尝试同时使用“startswith”和“next”命令，但没有奏效。还有其他方法吗？我还发送了我正在尝试使用的代码：
timestep = []
with open(file,'r') as f:
lines = f.readlines()
for line in lines:
line = line.split()
if line[0].startswith("ITEM: timestep"):
timestep.append(next(line))
print(timestep)
解决方法
逻辑是决定是否将当前的line附加到timestep。因此，您需要的是一个变量，当该变量为 TRUE 时，它会告诉您附加当前的 line。
timestep = []
append_to_list = False # decision variable
with open(file,'r') as f:
lines = f.readlines()
for line in lines:
line = line.strip() # remove "\n" from line
if line.startswith("ITEM"):
# Update add_to_list
if line == 'ITEM: TIMESTEP':
append_to_list = True
else:
append_to_list = False
else:
# append to list if line doesn't start with "ITEM" and append_to_list is TRUE
if append_to_list:
timestep.append(line)
print(timestep)
输出：
['253400','253500']
所以你的代码的问题很微妙。您有一个可迭代的列表 lines，但不能在列表上调用 next。
相反，把它变成一个显式迭代器，你应该没问题
timestep = []
with open(file,'r') as f:
lines = f.readlines()
lines_iter = iter(lines)
for line in lines_iter:
line = line.strip() # removes the newline
if line.startswith("ITEM: TIMESTEP"):
timestep.append(next(lines_iter,None)) # the second argument here prevents errors
# when ITEM: TIMESTEP appears as the
# last line in the file
print(timestep)
我也不确定你为什么包含 line.split，这似乎是不正确的（在任何情况下 line.split()[0].startswith('ITEM: TIMESTEP') 永远不可能是真的，因为拆分会将 ITEM: 和 {{ 1}} 到结果列表的单独元素中。）
要获得更可靠的答案，请考虑根据行以 TIMESTEP 开头的时间对数据进行分组。
ITEM这将让您传入整个文件，并为每个 def process_file(f):
ITEM_MARKER = 'ITEM: '
item_title = '(none)'
values = []
for line in f:
if line.startswith(ITEM_MARKER):
if values:
yield (item_title,values)
item_title = line[len(ITEM_MARKER):].strip() # strip off the marker
values = []
else:
values.append(line.strip())
if values:
yield (item_title,values)
组懒惰地生成一组值。然后你可以以某种合理的方式进行聚合。
ITEM: <whatever>首先 - 我不喜欢这个，因为它不能扩展。您只能很好地获得紧随其后的第一行，其他任何事情都只是呃......
但是你问了，所以...... import seaborn as sns
sns.lineplot(data=data,x='date',y='value',hue='type')
将在行上创建一个迭代器并使用它来保持位置。您无权访问该迭代器，因此 import matplotlib.pyplot as plt
fig,ax = plt.subplots()
for label,group in data.groupby('type'):
group.plot(x='date',label=label,ax=ax)
不会是您期望的下一个元素。但是您可以制作自己的迭代器并使用它：
for x in lines但是，如果您想扩展它... next 不是迭代这样的文件的好方法。您想知道下一行/上一行中的内容。我建议使用 lines_iter = iter(lines)
for line in lines_iter:
# whatever was here
timestep.append(next(line_iter))
:
for通过这种方式，您可以将其扩展为不同类型的可变长度项目。
,您可以使用 enumerate 来帮助进行索引引用。我们可以检查字符串 ITEM: TIMESTEP 是否在前一行，然后将整数添加到我们的时间步长列表中。
timestep = []
with open('example.txt','r') as f:
lines = f.readlines()
for i,line in enumerate(lines):
if "ITEM: TIMESTEP" in lines[i-1]:
timestep.append(int(line.strip()))
print(timestep)