CSV到字典，标题作为键，其余行作为Python中的字符串（值）列表

Question

我有一个这样的 CSV：

(row 0)    key1   key2   key3
(row 1)    A      D      G
(row 2)    B      E      H
(row 3)    C             I

我想阅读它并将信息放入字典中，如下所示：dict = {key1: ["A", "B", "C"], key2: ["D", "E"], key3: ["G", "H", "I"]}。

所以基本上我希望将第一行作为键，将每列的所有值作为列表的值（列表是字典中键的值）。

谢谢！

Answer 1

您可以 stack 和 groupby 将值聚合到 list，然后使用 .to_dict() 转到字典：

df = pd.read_csv("my_csv.csv")
df.stack().groupby(level=1).agg(list).to_dict()
# {'key1': ['A', 'B', 'C'], 'key2': ['D', 'E'], 'key3': ['G', 'H', 'I']}