从字典中的项目创建自定义的 pandas 数据框

Question

我想从字典中的项目创建一个自定义的 pandas DataFrame。例如，我可以使用 pandas 的 from_dict() 函数将字典转换为 DataFrame：

data = {'wordA': [3, 2, 1, 0], 'wordB': [33, 12, 1, 8], 'wordC': [54, 10, 7, 3]} 
pd.DataFrame.from_dict(data, orient='index', columns=['col1', 'col2', 'col3', 'col4'])

生成如下所示的 DataFrame：

        col1    col2    col3    col4
wordA   3       2       1       0
wordB   33      12      1       8
wordC   54      10      7       3

但是我想要的是只有 2 列，如下所示，其中word 列返回字典键，<​​strong>count 列返回列表中包含的字典值的计数。

word    count
wordA   4
wordB   4
wordC   4

我怎样才能实现这个目标？

Answer 1

您可以更改传递给构造函数的字典：

out = pd.DataFrame.from_dict({i: {'word': k, 'count': len(v)} 
                              for i, (k, v) in enumerate(data.items())}, 
                             orient='index')

或

out = (pd.DataFrame.from_dict({'count': {k: len(v) for k, v in data.items()}})
       .rename_axis('word').reset_index())

输出：

    word  count
0  wordA      4
1  wordB      4
2  wordC      4