简体   繁体   中英

How to use a txt file in .map() function in Pandas

I am creating a dictionary and using it in.map() function in Pandas, in the same code file. However, I want to store the dictionary in different txt file, and then call it in.map() function of the code. Is there any way to achieve this?

Code:

Hotels_Region = {"La Pizzeria" : "Italy",
"Domino's" : "Germany",
"Pizza Hut" : "Austria"}

df['country'] = df.store.map(Hotels_Region)

I want to store below code in different file:

Hotels_Region = {"La Pizzeria" : "Italy",
"Domino's" : "Germany",
"Pizza Hut" : "Austria"}

and then call it in

df['country'] = df.store.map(Hotels_Region)

You can do it by using Json

import json
#SAVE
with open('data.txt', 'w') as outfile:
    json.dump(Hotels_Region, outfile)

#LOAD
with open('data.txt') as json_file:
    data = json.load(json_file)

data

{'La Pizzeria': 'Italy', "Domino's": 'Germany', 'Pizza Hut': 'Austria'}

It will work in the map function

df=pd.DataFrame({'store':['La Pizzeria']})
df.store.map(data)

0    Italy

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM