重命名列后获取keyerror

时间:2017-04-08 08:17:52

标签: pandas numpy multiple-columns rename

我有df

df = pd.DataFrame({'a':[7,8,9],
                   'b':[1,3,5],
                   'c':[5,3,6]})

print (df)
   a  b  c
0  7  1  5
1  8  3  3
2  9  5  6

然后按this重命名第一个值:

df.columns.values[0] = 'f'

一切似乎都很好:

print (df)
   f  b  c
0  7  1  5
1  8  3  3
2  9  5  6

print (df.columns)
Index(['f', 'b', 'c'], dtype='object')

print (df.columns.values)
['f' 'b' 'c']

如果选择b,则效果很好:

print (df['b'])
0    1
1    3
2    5
Name: b, dtype: int64

但如果选择a,则会返回列f

print (df['a'])
0    7
1    8
2    9
Name: f, dtype: int64

如果选择f获取keyerror。

print (df['f'])
#KeyError: 'f'

print (df.info())
#KeyError: 'f'

有什么问题?有人可以解释一下吗?还是虫子?

1 个答案:

答案 0 :(得分:16)

您不应该更改values属性。

尝试df.columns.values = ['a', 'b', 'c'],然后获得:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-61-e7e440adc404> in <module>()
----> 1 df.columns.values = ['a', 'b', 'c']

AttributeError: can't set attribute

那是因为pandas检测到您正在尝试设置属性并阻止您。

但是,它无法阻止您更改基础values对象本身。

当您使用rename时,pandas会跟进一堆清理内容。我已粘贴下面的来源。

最终你所做的就是在不启动清理的情况下改变数值。您可以通过对_data.rename_axis的后续调用自行启动它(示例可以在下面的源代码中看到)。这将强制清理运行,然后您可以访问['f']

df._data = df._data.rename_axis(lambda x: x, 0, True)
df['f']

0    7
1    8
2    9
Name: f, dtype: int64

故事的道德:以这种方式重命名专栏可能不是一个好主意。

但是这个故事更奇怪

这很好

df = pd.DataFrame({'a':[7,8,9],
                   'b':[1,3,5],
                   'c':[5,3,6]})

df.columns.values[0] = 'f'

df['f']

0    7
1    8
2    9
Name: f, dtype: int64

这是罚款

df = pd.DataFrame({'a':[7,8,9],
                   'b':[1,3,5],
                   'c':[5,3,6]})

print(df)

df.columns.values[0] = 'f'

df['f']
KeyError:

事实证明,我们可以在显示values之前修改df属性,它显然会在第一个display上运行所有初始化。如果您在更改values属性之前显示它,则会出错。

weirder仍然

df = pd.DataFrame({'a':[7,8,9],
                   'b':[1,3,5],
                   'c':[5,3,6]})

print(df)

df.columns.values[0] = 'f'

df['f'] = 1

df['f']

   f  f
0  7  1
1  8  1
2  9  1

好像我们还不知道这是一个坏主意......

来源rename

def rename(self, *args, **kwargs):

    axes, kwargs = self._construct_axes_from_arguments(args, kwargs)
    copy = kwargs.pop('copy', True)
    inplace = kwargs.pop('inplace', False)

    if kwargs:
        raise TypeError('rename() got an unexpected keyword '
                        'argument "{0}"'.format(list(kwargs.keys())[0]))

    if com._count_not_none(*axes.values()) == 0:
        raise TypeError('must pass an index to rename')

    # renamer function if passed a dict
    def _get_rename_function(mapper):
        if isinstance(mapper, (dict, ABCSeries)):

            def f(x):
                if x in mapper:
                    return mapper[x]
                else:
                    return x
        else:
            f = mapper

        return f

    self._consolidate_inplace()
    result = self if inplace else self.copy(deep=copy)

    # start in the axis order to eliminate too many copies
    for axis in lrange(self._AXIS_LEN):
        v = axes.get(self._AXIS_NAMES[axis])
        if v is None:
            continue
        f = _get_rename_function(v)

        baxis = self._get_block_manager_axis(axis)
        result._data = result._data.rename_axis(f, axis=baxis, copy=copy)
        result._clear_item_cache()

    if inplace:
        self._update_inplace(result._data)
    else:
        return result.__finalize__(self)