将txt文件导入字典中的元组

Question

I have a data structure in a .txt file set up as follows:

andrew
3 3 1 0 3 0 3 0 0 -3 0 5 3 0 1 0 0 5 3 0 0 0 0 1 0 3 0 1 0 0 3 5 3 3 0 
0 0 5 0 5 0 3 3 0 -3 0 0 5 1 5 3 0 3 0 0 

I'm trying to import it as in this format:

{'user':[0,1,3,4,5]}

I've tried various implementations, but can't find anything to suit my needs and account for so many variables.

My current code is as follows:

    with open('ratings.txt', 'r') as f:
    for line in f:
        x = line.split()
        print(x)
        ratings[x[0]] = [y for y in x[1:]])

Any idea on how to improve the code, please?

Answer 1

you can read the whole file by read() method then split it so that you can create the dict as you want with out reading line by line

ratings = {}
with open('ratings.txt', 'r') as f:
    X = f.read().split()

    x = set([int (i) for i in X[1:] if int(i) >= 0]) # this will create a set which values are >= 0

    ratings[X[0]] = list(x) 


print(ratings)# output --> {'andrew': [0, 1, 3, 5]}

this is assuming your text file will always be in the format you mention above only one user at a time

As per Your text File I think you can do like this : you read 2 lines by 2 to until you find a empty line or the end of file like below

ratings = {}
with open('ratings.txt', 'r') as f:
    while True:
        name = f.readline().strip()
        values = f.readline().strip().split()
        #print(values)
        if not name: break
        else:
            ratings[name] = list(set(int (i) for i in values if int(i) >= 0))

print(ratings)# output --> {'andrew': [0, 1, 3, 5]}