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