如何在 Pandas 中使用 txt 文件 in.map() function

Question

我正在创建字典并在同一代码文件中的 Pandas 中使用它 in.map() function。 但是，我想将字典存储在不同的txt文件中，然后在代码的.map() function中调用它。 有什么办法可以做到这一点？

代码：

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

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

我想将以下代码存储在不同的文件中：

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

然后调用它

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

Answer 1

您可以使用 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'}

它将在 map function 中工作

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

0    Italy