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

时间:2018-09-06 14:44:31

标签: python pandas dataframe

我想将大熊猫数据框中的前50个项目转换为一个列表,该列表对于数据框中的每个索引都将具有值。即使数据框在该索引中没有任何值,我也希望列表的值为0。 例如,如下所示的熊猫数据框:

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,因为在熊猫数据框中找不到ID。

2 个答案:

答案 0 :(得分:3)

如果我正确理解:

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

答案 1 :(得分:2)

IIUC:

rename