I needed help to add into directory below text file. Can anyone can help me to do it? I tried
I have data.txt like below:?
A1234 161
A1234 106
A456 185
A456 108
037 125
**Output:**
directory = {
"A1234": [161,106],
"A456": [185,108],
"037": [125],
}
Thank you in advance for your help.
data.txt:
A1234 161
A1234 106
A456 185
A456 108
037 125
From file to dictionary:
with open('data.txt', 'r') as file:
data_lines = file.readlines()
directory = {}
for line in data_lines:
a, *b = line.split()
# convert all elements of b into integers:
b = [int(item) for item in b]
if directory.get(a, False):
if isinstance(b, list):
directory[a].extend(b)
else:
directory[a].append(b)
else:
directory[a] = list(b)
print(directory)
# {'A1234': [161, 106], 'A456': [185, 108], '037': [125]}
# prettified:
"""
{
'A1234': [161, 106],
'A456': [185, 108],
'037': [125]
}
"""
ValueError: too many values to unpack (expected 2) has nothing to do with the amount of data. It has to do with the line a, b = line.split() : it's unpacking the line split (a list) in two variables. Does your data file contain lines with more than 2 'words' ? (I guess so. If so, show me an exemple I will adapt the code) See more info hereA1234 161 55 and that's throwing the same error. I edited my answer, now it should work fine with any amount of 'words' or 'numbers' per line :)Try this:
output_dict = {}
data = list(map(
lambda a:a.strip().split(),
open("data.csv").readlines()
))
for k,v in data:
try:
output_dict[k].append(v)
except:
output_dict[k] = [v]
output_dict
Output:
{'A1234': ['161', '106'], 'A456': ['185', '108'], '037': ['125']}
Here's a pandas solution. First we employ read_csv() and use one or more spaces as our delimiter. We need to specify string (object) dtypes to get string values for the list items as in your output. If you want ints, you could skip the dtype argument and pandas will infer them. Next we groupby the fist column (0) and convert the values in column 1 to a list with apply. Finally, we use .to_dict to get a dictionary.
import pandas as pd
df = pd.read_csv('dir.txt', header=None, sep=r"[ ]{1,}", dtype='object')
directory = df.groupby(0)[1].apply(list).to_dict()
output:
{'037': ['125'], 'A1234': ['161', '106'], 'A456': ['185', '108']}
