使用字典映射数据帧索引

时间:2017-04-11 21:44:33

标签: python pandas

为什么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
'''

6 个答案:

答案 0 :(得分:17)

我没有回答你的问题......只是给你一个更好的解决方法 使用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

答案 1 :(得分:15)

在末尾添加get

df['Two']=df.index.map(map_dict.get)
df
Out[155]: 
   one    Two
A   10  every
B   20   good
C   30    boy
D   40   does
E   50   fine

答案 2 :(得分:5)

调用map的另一种解决方法:

df['two'] = pd.Series(map_dict)

df

   one    two
A   10  every
B   20   good
C   30    boy
D   40   does
E   50   fine

在任何情况下,在映射问题得到解决之前(根据juanpa.arrivillaga的评论),您必须将索引或dict-to-map转换为pandas系列。

答案 3 :(得分:2)

自熊猫0.23.x版本(2018年5月15日发布)开始,此问题已修复:

import pandas as pd
pd.__version__        # 0.23.4

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.index.map(map_dict)
#        one
# every   10
# good    20
# boy     30
# does    40
# fine    50

What's New page中获取大熊猫0.23.0:

  

Index.map()现在可以接受系列和字典输入对象(GH12756,GH18482,GH18509)。

有关更多信息,请查看Index.map的帮助页面

答案 4 :(得分:1)

map(一个python关键字)显然被用作df.index的方法

因为它有自己的内部需求,所以不允许传递一个没有__call__方法的参数。

lambda并且函数可以调用,这是一个简单的测试:

def foo():
    pass
if foo.__call__:
    print True
# Prints True

bar = lambda x: x+1
if bar.__call__:
    print True
# Prints True

print {'1':'one'}.__call__
# AttributeError: 'dict' object has no attribute '__call__'

答案 5 :(得分:1)

更短的选择 - 没有明确调用to_seriespd.Series

df['two'] = df.rename(map_dict).index