python - 从 Pandas 数据框列生成列表(包括缺失值)

Question

这是我正在使用的简单 Pandas 数据框:

>>> df2
    Name  Day  Score
0  Allen    1     25
5  Allen    3      9

我想要做的是生成一个 Allen 分数列表，并在从 1 到 5 的所有日子都没有分数的日子里发出信号。这是我能够做到的:

>>> [df2[df2['Day']==i]['Score'].values[0] if i in list(df2['Day']) else None for i in range(1,6)]
[25, None, 9, None, None]

肯定有更简单的方法来做到这一点？ 这是什么？

在相关说明中，我想为更大数据框中的每个人执行此操作。最好的方法是什么？现在我能想到的就是将 apply 与上面的表达式一起使用，但这看起来比应该的更笨拙。感谢您的任何建议!

Answer 1

我会先将索引设置为“Day”并重新编制索引:

>>> df2 = df2.set_index('Day').reindex(range(1,6))

      Name  Score
Day              
1    Allen     25
2      NaN    NaN
3    Allen      9
4      NaN    NaN
5      NaN    NaN

从那里你有很多选择可以进一步进行。如果您只想要一个列表:

>>> df2['Score'].tolist()

[25.0, nan, 9.0, nan, nan]

您可能需要查看 fillna() 以了解指定缺失值的不同选项。