change 2d array by a "dictionary translation" in python

Question

I have the following 2d array [[2,3,4],[4,4,2] and the following dictionary {2:7,3:5,4:6}
I would like somehow to transform the array using the dictionary, i.e. have as an output the following result:

[[7,5,6],[6,6,7]]

Is there a simple (maybe built-in function) to do so?

I don't think there is an inbuilt function to do exactly what you want. — Carlos Gonzalez, Jun 12 '19 at 10:05
Tip to get a good answer: Please do add your previously written code to the question. Also, you missed a closing bracket in `[[2,3,4],[4,4,2]`. — Mazdak, Jun 12 '19 at 10:05

score 1 · Accepted Answer · answered Jun 12 '19 at 10:06

1

You can use np.vectorize():

x = np.array([[2,3,4],[4,4,2]])
y = {2:7,3:5,4:6}
np.vectorize(y.get)(x)

array([[7, 5, 6],
       [6, 6, 7]])

answered Jun 12 '19 at 10:06

zipa

27,316
6
40
58

change 2d array by a "dictionary translation" in python

1 Answers1