时间戳

首页 > 解决方案 > 有效地将大熊猫数据框转换为列表

python - 有效地将大熊猫数据框转换为列表

问题描述

我想将大熊猫数据框中的前 50 个项目转换为一个列表，对于数据框中的每个索引，该列表将具有该值。即使数据框在该索引中没有任何值，我也希望列表的值为 0。例如，看起来像这样的 pandas 数据框：

ID      Count
0       20
1       50
2       60
4       90
5       20
.
49      65
.
9999999 60054

将转换为以下列表，只有数据框的前 50 个元素是相关的：

[20, 50, 60, 0, 90, 20......,65]

请注意，在 index=3 处，列表中的值为 0，因为在 pandas 数据框中找不到 ID。

标签： pythonpandasdataframe

解决方案

如果我理解正确：

mylist = (df.iloc[:50].set_index('ID')
          .reindex(range(50), fill_value=0)['Count']
          .tolist())

推荐阅读