使用字典映射数据框索引
发布于 2021-01-29 17:47:16
为什么df.index.map(dict)
不像df['column_name'].map(dict)
?
这是一个尝试使用index.map的小例子:
import pandas as pd
df = pd.DataFrame({'one': {'A': 10, 'B': 20, 'C': 30, 'D': 40, 'E': 50}})
map_dict = {'A': 'every', 'B': 'good', 'C': 'boy', 'D': 'does', 'E': 'fine'}
df
'''
one
A 10
B 20
C 30
D 40
E 50
'''
df['two'] = df.index.map(mapper=map_dict)
这引起了 TypeError: 'dict' object is not callable
给它喂lambda可以工作:
df['two'] = df.index.map(mapper=(lambda x: map_dict[x])); df
'''
one two
A 10 every
B 20 good
C 30 boy
D 40 does
E 50 fine
'''
但是,重置索引和列上的映射可以按预期工作,而不会产生抱怨:
df.reset_index(inplace=True)
df.rename(columns={'index': 'old_ndx'}, inplace=True) #so there's no index name confusion
df['two'] = df.old_ndx.map(map_dict); df
'''
old_ndx one two
0 A 10 every
1 B 20 good
2 C 30 boy
3 D 40 does
4 E 50 fine
'''
关注者
0
被浏览
53
1 个回答
-
我没有回答您的问题…只是为您提供更好的解决方法。
使用to_series()
它们map
df = pd.DataFrame({'one': {'A': 10, 'B': 20, 'C': 30, 'D': 40, 'E': 50}}) map_dict = {'A': 'every', 'B': 'good', 'C': 'boy', 'D': 'does', 'E': 'fine'} df['two'] = df.index.to_series().map(map_dict) df one two A 10 every B 20 good C 30 boy D 40 does E 50 fine