尝试将列设置为数据帧中的索引时出错

Question

我有以下代码：

A = pd.DataFrame([[1, 2], [1, 3], [4, 6]], columns=[['att1', 'att2']])
A['idx'] = ['a', 'b', 'c']
A

这工作正常，直到我这样做（试图设置列'idx'作为索引的数据帧）

A.set_index('idx', inplace=True)

这会引发一个错误

TypeError: only integer scalar arrays can be converted to a scalar index

这是什么意思？

Answer 1

错误是当您创建A

columns = [['att1', 'att2']]

如果打印A.columns，您将获得：

MultiIndex([('att1',),
            ('att2',),
            ( 'idx',)],
           )

因此，“idx”实际上不在您的列中，您无法设置索引。现在，这将起作用：

A.set_index(('idx',))

并给予：

       att1 att2
(idx,)          
a         1    2
b         1    3
c         4    6

但是，您应该只使用以下方法来修复您创建的A：

columns = ['att1', 'att2']