Replacing parts of string from values in dict in python

Question

I have a dictionary in python e.g -

partitions = {'train': ['data/xyz/id-1', 'data/xyz/id-2', 'data/xyz/id-3',......], 
               'validation': ['data/pqr/id-4','data/pqr/id-5',.......]}

I want to replace 'data/xyz ' to 'samples/folder1' for all the values of the key 'train'.

the new values will be

samples/folder1/id-1, samples/folder1/id-2, samples/folder1/id-3 etc

similarly , 'data/pqr' to 'samples/test' for all the values of the key 'validation'.

I am unable to figure out how to do it in a concise way. Any help will be appreciated.

Chris Doyle · Accepted Answer · 2021-06-07 09:27:05Z

2

These are just lists assigned to keys, so you could use a list comprehension to rebuild the list and replacing the values in the strings.

partitions = {
    'train': ['data/xyz/id-1', 'data/xyz/id-2', 'data/xyz/id-3'],
    'validation': ['data/pqr/id-4', 'data/pqr/id-5']
}

print(partitions)
partitions['train'] = [value.replace('data/xyz', 'samples/folder1') for value in partitions['train']]
partitions['validation'] = [value.replace('data/pqr', 'samples/test') for value in partitions['validation']]
print(partitions)

OUTPUT

{'train': ['data/xyz/id-1', 'data/xyz/id-2', 'data/xyz/id-3'], 'validation': ['data/pqr/id-4', 'data/pqr/id-5']}
{'train': ['samples/folder1/id-1', 'samples/folder1/id-2', 'samples/folder1/id-3'], 'validation': ['samples/test/id-4', 'samples/test/id-5']}

answered Jun 7, 2021 at 9:27

Chris Doyle

12.4k2 gold badges30 silver badges49 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Sandeep Pandey Over a year ago

Thanks! exactly what I was looking for. I wanted to avoid loops .

Mehdi Mostafavi · Accepted Answer · 2021-06-07 09:28:24Z

2

Using simple loop and replace for strings:

partitions = {'train': ['data/xyz/id-1', 'data/xyz/id-2', 'data/xyz/id-3'], 
               'validation': ['data/pqr/id-4','data/pqr/id-5']}
replaces = {'train' : ['samples/folder1', 'data/xyz'] , 'validation' : ['samples/test','data/pqr']}
for key , value in partitions.items():
    temp = []
    for v in value:
        temp.append(v.replace(replaces[key][1], replaces[key][0]))
    partitions[key] = temp
print(partitions)

answered Jun 7, 2021 at 9:28

Mehdi Mostafavi

8911 gold badge13 silver badges28 bronze badges

Collectives™ on Stack Overflow

Replacing parts of string from values in dict in python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related