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)