CODEWITHSUNDEEP
pythonlist-comprehensiondata-munging
JackWM
JackWM

Reputation: 10535

Python program that generate a list from a given list according to a mapping

E.g.

org_list : 

aa b2 c d

mapping :

aa 1
b2 2
d 3
c 4

gen_list:

1 2 4 3

What is the Python way to implement this? Suppose org_list and the mapping are in files org_list.txt and mapping.txt, while the gen_list will be written into gen_list.txt

Btw, which language do you expect is very simple to implement this?

Upvotes: 1

Views: 95

Answers (5)

Aleksei Zyrianov
Aleksei Zyrianov

Reputation: 2342

Here's a solution for your case.

with open('org_list.txt', 'rt') as inp:
    lines = inp.read().split()
    org_list = map(int, lines)

with open('mapping.txt', 'rt') as inp:
    lines = inp.readlines()
    mapping = dict(line.split() for line in lines)

gen_list = (mapping[i] for i in org_list) # Or you may use `gen_list = map(mapping.get, org_list)` as suggested in another answers

with open('gen_list.txt', 'wt') as out:
    out.write(' '.join(gen_list))

I think Python handles this situation gracefully enough.

Upvotes: 4

Rushy Panchal
Rushy Panchal

Reputation: 17532

mapping = dict(zip(org_list, range(1, 5)))       # change range(1, 5) to whatever
gen_list = [mapping[elem] for elem in org_list]  # you want it to be

Upvotes: 0

Martijn Pieters
Martijn Pieters

Reputation: 1121814

Just loop through the list with a list comprehension:

gen_list = [mapping[i] for i in org_list]

Demo:

>>> org_list = ['aa', 'b2', 'c', 'd']
>>> mapping = {'aa': 1, 'b2': 2, 'd': 3, 'c': 4}
>>> [mapping[i] for i in org_list]
[1, 2, 4, 3]

If you have this data in files, first build the mapping in memory:

with open('mapping.txt') as mapfile:
    mapping = {}
    for line in mapfile:
        if line.strip():
            key, value = line.split(None, 1)
            mapping[key] = value

then build your output file from the input file:

with open('org_list.txt') as inputfile, open('gen_list.txt', 'w') as outputfile:
    for line in inputfile:
        try:
            outputfile.write(mapping[line.strip()] + '\n')
        except KeyError:
            pass  # entry not in the mapping

Upvotes: 5

Raymond Hettinger
Raymond Hettinger

Reputation: 226296

Try using map() or a list comprehension:

>>> org_list = ['aa', 'b2', 'c', 'd']
>>> mapping = {'aa': 1, 'b2': 2, 'd': 3, 'c': 4}

>>> map(mapping.__getitem__, org_list)
[1, 2, 4, 3]

>>> [mapping[x] for x in org_list]
[1, 2, 4, 3]

Upvotes: 1

Katriel
Katriel

Reputation: 123632

Another way:

In [1]: start = [1,2,3]
In [2]: mapping = {1: "one", 2: "two", 3: "three"}
In [3]: map(mapping.get, start)
Out[3]: ['one', 'two', 'three']

Upvotes: 3

Related Questions