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

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']