我有一个CSV文件,我正在尝试使用python提取数据。没有得到正确的结果。有人能帮忙吗?代码和样本数据随附

我有一个CSV文件,我正在尝试使用python提取数据。没有得到正确的结果。有人能帮忙吗?代码和样本数据随附,python,csv,Python,Csv,简单的python代码打开一个文件,附上示例,并提取对应于各自数据库中所有用户的所有行。但是,它不能正常工作。请注意,该文件是根据用户排序的 queries_file = file("queries_new_distinct_sorted.csv") old_user = None current_user = None user_file = None queries = "" # variable to capture all the rows as \n separated string

简单的python代码打开一个文件,附上示例,并提取对应于各自数据库中所有用户的所有行。但是,它不能正常工作。请注意,该文件是根据用户排序的

queries_file = file("queries_new_distinct_sorted.csv")
old_user = None
current_user = None
user_file = None
queries = "" # variable to capture all the rows as \n separated string
for line in queries_file:
    query, user, timestamp, categories = line.strip().split(",")
    # to extract component from the csv file

    current_user = user

    if old_user == None:
        old_user = current_user         
        queries += timestamp +","+ categories+"\n"

    elif old_user != current_user:
        if len(queries.strip().split('\n')) > 3:            
            user_file = open("users/" + old_user,'w+')
            user_file.write(queries)
            user_file.close() 
            old_user = current_user
            queries = timestamp +","+ categories+"\n"

    else:       
        queries += timestamp +","+ categories+"\n"


queries_file.close()
文件:

ipd,6086533,2006-05-12_19:45:23,shopping/vehicles/parts and accessories
tire rack,6086533,2006-05-13_22:29:44,shopping/vehicles/parts and accessories
volvo r 70 speed parts,6086533,2006-05-14_23:04:33,shopping/vehicles/autos
puerto rico,6086589,2006-03-07_21:39:36,travel and tourism/travel guides
espn spanish,6086589,2006-03-12_19:53:26,world/español/medios de comunicación
发现了问题。
如果查询长度小于2,我没有重置值。

您遇到的实际问题是什么?预期输出是什么?预期输出是:同一用户的所有事务都应放入其用户id的文件中。问题是:它没有捕获所有用户。