将文本文件中的键和值添加到字典

Question

I have a text file test.txt with two lines on it:

1 "test one"
2 "test two"

I just want to add those 2 lines to test_dict :

test_dict = {} 
with open("test.txt", "r") as f: 
    for line in f:
        (key, val) = line.split()
        test_dict[int(key)] = val

print(test_dict)

Error:

ValueError: too many values to unpack (expected 2)

Expected result:

test_dict = {1: "test one", 2: "test two"}

Answer 1

Your code above should work, so your main issue may be that you have a different text file to what you attached.

For example, if I change test.txt to the following:

1 test_one
2 test_two a

I get the same error as seen in your original question.

One solution to this, if you're using an unknown input, is to use the index of the array instead of trying to unpack it into separate variables. Therefore, if you have more than 2 pieces of data on a line, it'll run while ignoring the seperate errors:

test_dict = {} 

# Open the file
with open("test.txt", "r") as file: 
    # Return an iterable of each line of the file
    lines = file.readlines()
    # Iterate through the lines
    for line in lines:
        # Split the line using the ' ' (space) character
        result = line.split(' ')
        # Set the value in the dictionary to the value
        test_dict[int(result[0])] = result[1]

print(test_dict)

If, instead you want to incorporate error handling, you can check the length of the line, and print out an error message. eg:

test_dict = {} 

# Open the file
with open("test.txt", "r") as file: 
    # Return an iterable of each line of the file
    lines = file.readlines()
    # Iterate through the lines
    for no, line in enumerate(lines):
        # Split the line using the ' ' (space) character
        result = line.split(' ')
        if len(result) != 2:
            # ERROR!
            print(f"Line {no+1}: '{line}' has an invalid number of elements")
        else:
            # Set the value in the dictionary to the value
            test_dict[int(result[0])] = result[1]

print(test_dict)

This outputs:

Line 2: '2 test_two a' has an invalid number of elements
{1: 'test_one\n'}

And, as you can see, it ignores the line with an error.

Answer 2

After you edited your question, this should work:

test_dict = {}
with open("testtt.txt", "r") as f:
    for line in f:
        (key, val) = line.split(maxsplit=1)
        test_dict[int(key)] = val

print(test_dict)

Because we need the first item as a key, and "rest of them" as value, so I pass maxsplit=1 as an argument to split() . This only split the line only 1 time from the left.