如图所示,每组数据都有不同的行数,只判断每组数据的第三行是否大于2,如果有此组数据中第三列数字任意一个数字大于2,就把整租数据以原格式存放在一个新文件中,如果没有一个大于2的,则整个数组都舍掉,这个过程该怎么实现呢?望不吝指教!
ff = """666 333 3
1 2 3
3 4 5
3 4 5
666 333 3
1 2 3
3 4 5
3 4 5
666 333 3
1 2 3
3 4 5
3 4 5
666 333 3
1 2 0
3 4 0
3 4 0
"""
import re
infoList = re.split('(666.*\n)', ff)
result = []
for info in infoList:
if info:
if info.startswith('666'):
result.append(info)
else:
splitInfo = info.split('\n')
if any(True for t in splitInfo if len(t.split(' ')) > 2 and int(t.split(' ')[2]) > 2):
result.append(info)
print(''.join(result))
用pandas