从一个文件中选择使用某些条件的行

Question

我是Python新手，我需要帮助：

我有一个像这些行的文件（见下文）。我想复制此文件中的行，并且由于某些行具有相似的日期，因此如果行具有相似的日期，我想选择具有结束列表的最小值的行。

'1990-01-01','f','2' 
'1990-01-02','c','4' 
'1990-01-01','j','2.5' 
'1990-01-01','j','1.5' 
'1990-01-02','b','3.9' 

Answer 1

一些代码第一：

text_data = """'1990-01-01','f','2' 
'1990-01-02','c','4' 
'1990-01-01','j','2.5' 
'1990-01-01','j','1.5' 
'1990-01-02','b','3.9' 
""" 

import csv 
from StringIO import StringIO 
from operator import itemgetter 
from itertools import groupby 


temp = StringIO(text_data) 
tabular = list(csv.reader(temp, quotechar="'")) 
tabular.sort(key=lambda L: (L[0], float(L[2])) 

for key, val in groupby(tabular, itemgetter(0)): 
    print next(val) # or do something appropriate 

我已经把代码中的文本，并包裹在一个StringIO对象来模拟一个文件，但通常人们会打开使用CSV文本。读者在使用open（）的文件上。

然后数据按数据列（索引0）和值列（索引2）排序 - 这种排序顺序确保相同的日期在一起（我意识到你提到'相似'，但这个配方可以是适应），并且对于每个日期，最低值列首先出现。

itertools.groupby（）用于遍历符合某些条件的键 - 它们符合条件时，它们位于同一个“组”中。 'key'（分组的内容）将是日期的值，'val'将是该键中可用值的另一个迭代。由于只需要第一个，所以使用next（val）将返回最小值的结果。

输出是：

['1990-01-01', 'j', '1.5'] 
['1990-01-02', 'b', '3.9']