python – pandas使用loc插入行的Dataframe如果第一个参数不是列表,则会引发错误错误？

2023年5月30日 230次阅读

这是问题的最小例子：

import pandas as pd
example = pd.DataFrame(index=pd.np.arange(2) , columns=['A', 'B', 'C']).astype('object')
example.loc[0] = [['a'], 'b', [1,2,3]]
example.loc[1] = ['a', 'b', [1,2,3]]

我得到一个ValueError：用序列设置一个数组元素

以下是spyder中DataFrame的样子：
《python – pandas使用loc插入行的Dataframe如果第一个参数不是列表,则会引发错误错误？》

我不明白为什么第一个元素必须“嵌套”.
为什么第二行不起作用？我究竟做错了什么？
请查看可以直接插入’b’的第二列.

我投票给了答案,我个人最喜欢提供的所有三种工作解决方案.我发现它比其他答案更加pythonic,而且它也是最快的.有关速度比较,请参阅this post

最佳答案 你正在推动熊猫的界限.它不擅长处理更高级别的对象.所以我们要小心.

在这种情况下,Pandas不会立即看到它是一个对象数组,当它到达序列时就会失败.

解决

包裹在一个系列对象中

import pandas as pd
example = pd.DataFrame(index=pd.np.arange(2) , columns=['A', 'B', 'C']).astype('object')
example.loc[0] = [['a'], 'b', [1,2,3]]
example.loc[1] = pd.Series(['a', 'b', [1,2,3]], example.columns)

example

     A  B          C
0  [a]  b  [1, 2, 3]
1    a  b  [1, 2, 3]
点赞