python - 令人沮丧的错误：在 Pandas 中创建新列时抛出 KeyError

Question

这很奇怪，我以前使用过这段代码并且它有效。但现在它给我一个 KeyError 说：

KeyError：“[索引（['0'，'1']，dtype ='object'）]都不在[列]中”

d = {'col1': [1, 2, 3], 'col2': [3, 4, 5]}
df = pd.DataFrame(data=d)
df[[str(c) for c in range(2)]] = [[5,6],[6,6], [7,7]]

我什至让我的朋友在他身边测试它，他执行代码没有任何错误没有问题。还有，我的Pandas版本确实不一样，一个是1.1.4，一个是1.0.3。

另外，如果有人有一个优雅的解决方案，.loc或者.iloc请让我知道。当版本控制不是问题时，为什么这在我的计算机上不起作用，我真的很沮丧。

标签： pythonpandasnumpy

Answer 1

对我来说工作得很好。

另一种解决方案是创建新的 DataFrame 并添加到原来的：

df = df.join(pd.DataFrame([[5,6],[6,6], [7,7]], index=df.index, columns=['0','1']))
print (df)
   col1  col2  0  1
0     1     3  5  6
1     2     4  6  6
2     3     5  7  7