如何使用Python从文件的一部分制作整数列表？

Question

我有一个如下文件 ：

@ junk
...
@ junk
    1.0  -100.102487081243
    1.1  -100.102497023421
    ...   ...
    3.0  -100.102473082342
&
@ junk
...

我只对@和&字符之间的两列数字感兴趣。 这些字符可能会出现在文件中的其他任何位置，但永远不会出现在数字块中。

我想创建两个列表 ，一个带有第一列，一个带有第二列。

List1 = [1.0, 1.1,..., 3.0]
List2 = [-100.102487081243, -100.102497023421,..., -100.102473082342]

我一直在使用Shell脚本来为这些文件准备一个更简单的Python脚本，以创建列表，但是，我正在尝试将这些过程迁移到Python上以实现更一致的应用程序。 有任何想法吗？ 我对Python和文件处理的经验有限。

编辑：我应该提到，此数字块出现在文件中的两个位置。 两个数字块是相同的。

Edit2：通用函数对此最为满意，因为我将其放入自定义库中。

目前的努力

我目前使用shell脚本将数字块以外的所有内容修剪成两列。 从那里开始，使用以下功能对我来说是微不足道的

def ReadLL(infile):
    List = open(infile).read().splitlines()
    intL = [int(i) for i in List]
    return intL

通过我的主叫声

import sys
import eLIBc
infile = sys.argv[1]
sList = eLIBc.ReadLL(infile)

问题是知道如何使用Python而不是使用Shell脚本从原始文件中提取数字块。

Answer 1

您想遍历文件本身，并为找到没有 @字符的第一行设置一个标志，此后可以开始收集数字。 在一行上找到&字符时，请中断阅读。

def readll(infile):    
    with open(infile) as data:
        floatlist1, floatlist2 = [], []
        reading = False

        for line in data:
            if not reading:
                if '@' not in line:
                    reading = True
                else:
                    continue

            if '&' in line:
                return floatlist1, floatlist2

            numbers = map(float, line.split())
            floatlist1.append(numbers[0])
            floatlist2.append(numbers[1])

所以上面：

• 将'reading'设置为False ，并且仅当找到没有'@'的行时才将其设置为True 。
• 当'reading'为True ：
  • 如果该行包含&则返回读取的数据
  • 否则，假定该行包含两个用空格分隔的浮点值，并将它们添加到各自的列表中

通过返回，该函数结束，文件自动关闭。 仅读取第一个块，而忽略文件的其余部分。

Answer 2

试试看：

with open("i.txt") as fp:
    lines = fp.readlines()
    data = False
    List1 = []
    List2 = []
    for line in lines:
        if line[0] not in ['&', '@']:
            print line
            line = line.split()
            List1.append(line[0])
            List2.append(line[1])
            data = True
        elif data == True:
            break

print List1
print List2

这应该给您第一组数字。

输入：

@ junk
@ junk
1.0  -100.102487081243
1.1  -100.102497023421
3.0  -100.102473082342
&
@ junk
1.0  -100.102487081243
1.1  -100.102497023421

输出：

['1.0', '1.1', '3.0']
['-100.102487081243', '-100.102497023421', '-100.102473082342']

更新资料

如果您需要两个块，请使用以下命令：

with open("i.txt") as fp:
    lines = fp.readlines()
    List1 = []
    List2 = []
    for line in lines:
        if line[0] not in ['&', '@']:
            print line
            line = line.split()
            List1.append(line[0])
            List2.append(line[1])

print List1
print List2