如何使用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和文件處理方面的經驗有限。

編輯：我應該提到，這個數字塊出現在文件的兩個地方。這兩個數字塊是相同的。

編輯2：一般函數將是最令人滿意的，因爲我將它放到自定義庫中。

目前的努力

我目前使用的是shell腳本一切，但數塊修剪出成兩個單獨的列。從那裏，它是平凡的我從我的主要

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

問題稱這是使用下面的函數

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

是知道如何從原始文件與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]) 

所以上面：

• 套 '讀' 到False，並且僅當不'@'線被發現，是設定爲True。
• 「閱讀」時是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