python - 将重复列名列表和值列表转换为数据框

Question

假设我有两个列表。一个包含重复的标题列表。一个包含值。请参见下面的示例:

cols = ['col1', 'col2', 'col3', 'col1', 'col2', 'col3']
values = [1, 3, 4, 6, 7, 1]

我如何根据这种具有三列(“col1”、“col2”、“col3”)和两行的结构创建数据框。

想要的结果

'col1', 'col2', 'col3'
1,3,4
6,7,1

Answer 1

您可以尝试将 values 转换为 series，然后在 cols 上进行 groupby 并聚合为 list，最后转换为 dataframe 并转置

s = pd.Series(values).groupby(cols).agg(list)
pd.DataFrame.from_dict(s.to_dict())

---

   col1  col2  col3
0     1     3     4
1     6     7     1

或者使用默认字典:

from collections import defaultdict
d = defaultdict(list)
for k,v in zip(cols,values):
        d[k].append(v)
pd.DataFrame.from_dict(d)