Database 使用python为组中的每个元素添加一个序列号

发布于03月31日

我有一个数据框，每个人都有多条记录.我想按顺序枚举Python中每个个体的记录.实际上，我想在下表中创建‘SEQUENCE’列:

patient  date      sequence
145      20Jun2009        1
145      24Jun2009        2
145      15Jul2009        3
582      09Feb2008        1
582      21Feb2008        2
987      14Mar2010        1
987      02May2010        2
987      12May2010        3

这个问题与here基本相同，但我正在使用python，无法实现sql解决方案.我怀疑我可以使用一个可计算计数的groupby语句，但到目前为止一直没有成功.谢谢

推荐答案

问题是如何对多列数据进行排序.

一个简单的技巧是将key参数用于sorted函数.

您将按从数组的列构建的字符串进行排序.

rows = ...# your source data

def date_to_sortable_string(date):
  # use datetime package to convert string to sortable date.
  pass

# Assume x[0] === patient_id and x[1] === encounter date

# Sort by patient_id and date
rows_sorted = sorted(rows, key=lambda x: "%0.5d-%s" % (x[0], date_to_sortable_string(x[1])))

for row in rows_sorted:
  print row