import time
import numpy as np
t1=time.time()
dict1={}
for i in range(1,4935):
  dict1[i]=[]
with open('/storage/st2/aiqiyi/train_v2.txt','r') as f:
    while True:
      lista=f.readline()  #和f.read（）有區別，這只是讀取一行   
      if not lista: #結束判斷句
        break
      l=int(lista[28:len(lista)-1])
      if l in dict1:
        dict1[l].append(lista[0:28])
        #print('1')
      #print(l)
      #print(lista)
    f=open('/storage/st2/aiqiyi/trainidmerge.txt','a+')
    for i in dict1:
      m=str(i)
      f.write(m)  #輸出每個字典的鍵
      f.write(' ')
      for j in dict1[i]:  #把每個字典的鍵值輸出來
        n=str(j)+' '
        f.write(n)
      f.write('\n')
    print('t:',time.time()-t1)
 

儲存到檔案時為了防止覆蓋掉上一次儲存的內容，可以使用‘a+’

 f=open('/storage/st2/aiqiyi/trainidmerge.txt','a+')

import time
t1=time.time()
with open('/storage/st2/aiqiyi/val_v2.txt','r') as f:
  lista=f.readline()  #讀取一行資料，是一個整體的字串
  t=lista.split()     #去掉空格和tab返回的是陣列
  print(t)
  x=t.pop(0)          #去掉陣列的第一個元素
  print(x)
  print(t)
  


print('time:',time.time()-t1)