python - 有效地将大型 Pandas 数据框转换为列表

我想将一个 Pandas 数据框中的前 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。

最佳答案

如果我理解正确的话:

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

关于python - 有效地将大型 Pandas 数据框转换为列表，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/52206697/