python - 创建 Pandas DataFrame 的元素并将其设置为列表

Question

我有一个正在逐行创建的 Pandas DataFrame(我知道，我知道，它不是 Pandorable/Pythonic..)。我像这样使用 .loc 创建元素

output.loc[row_id, col_id]

我想将此值设置为一个空列表 []。

output.loc[row_id, col_id] = []

不幸的是，我收到一条错误消息，指出我的键和值的大小不匹配(Pandas 认为我正在尝试设置值 with 而不是 to 可迭代) .

有办法吗？

谢谢!

Answer 1

你需要确定两件事:

1. 该位置恰好有一个条目，
2. 列有 dtype 对象(实际上，在测试中这似乎不是问题)。

---

执行此操作的一种 hacky 方法是使用带有 [] 的系列:

In [11]: df = pd.DataFrame([[1, 2], [3, 4]], columns=['A', 'B'])

In [12]: df.loc[[0], 'A'] = pd.Series([[]])

In [13]: df
Out[13]:
    A  B
0  []  2
1   3  4

pandas 并不真的希望您使用 [] 作为元素，因为它通常效率不高并且会使聚合更加复杂(并且无法使用 cythonisable)。

---

一般来说，您不想逐个单元地构建 DataFrames，总有(几乎？)更好的方法。