无法在python循环中执行任务

2024-05-21 02:12:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我用日期和位置代码匹配两个csv文件中的ID字段,两个文件看起来都是这样的(当然有不同的ID):

ID   code   date
543  452   '1/4/2009'
543  452   '1/8/2009'
543  452   '2/1/2009'
543  452   '2/14/2009'
543  452   '3/6/2009'
543  452   '3/9/2009'
874  210   '1/4/2009'
874  210   '1/9/2009'
874  210   '1/24/2009'
874  210   '2/7/2009'

我将这些作为numpy数组加载,然后循环通过,但是我无法在循环中执行特定任务。我需要从我的第二个csv(vms)中创建一个子集数组,该数组包含与第一个csv(lb)中的记录匹配的所有记录。以下是我正在使用的代码:

import numpy as np
from collections import defaultdict

lb = np.genfromtxt('file2009.csv',dtype = None, delimiter = ',', names = True)
vms = np.genfromtxt('2009final.csv',dtype = None, delimiter = ',', names = True)

id_list = list(set(lb['ID']))

matchList = defaultdict(list)

for e in id_list:
    log = lb[lb['ID']==e]
    vms_list = vms[ np.logical_and(vms['date']==log[0]['date'] , vms['code']==log[0]['code'] )] 
    vms_id_list = list(set(vms_list['ID']))
    for row in log:
        if len(vms_id_list) == 1:
            break
        else:
            vmsIDs = vms[vms['ID'] == vms_id_list[0]]
            vms_ids = str(vms_id_list)
            for every in vms:
                if str(every['ID']) in vms_ids:
                    vmsIDs = np.hstack([vmsIDs, every])
            vms_id_list = []
            vms_list = vmsIDs[ np.logical_and(vmsIDs['date']==row['date'] , vmsIDs['code']==row['code'] )]
            vms_id_list = list(set(vms_list['ID']))
    matchList[e].append(vms_id_list)

当我运行代码时,不会产生任何错误,但很明显,行:
vms_list = vms[ np.logical_and(vms['date']==log[0]['date'] , vms['code']==log[0]['code'] )]
未正确运行,因为vms\u list为空,这会在vmsIDs = vms[vms['ID'] == vms_id_list[0]]行生成错误。 我真的很困惑,因为列表是创建的,它完全是空的,即使我知道有许多记录符合条件。
如果我简单地将代码行从循环中复制出来并独立运行它,它就可以正常工作,所以问题在于它在循环中执行。我不知道为什么这样做行不通,我甚至尝试过将语言从numpy逻辑单元改为:

for e in id_list:
    log = lb[lb['ID']==e]
    for each in vms:
        if each['date'] == log[0]['date'] and each['code'] == log[0]['code']:
            vms_id_list.append(each['ID'])

但它不起作用,列表是空的。这一步的重点是从vms中获取一个id列表,该列表与lb中一行的条件相匹配。但是它仍然不起作用,列表是空的。有什么锁我没看见吗?有没有别的方法来做我想做的事?你知道吗


Tags: csv代码inlogid列表fordate
1条回答
网友
1楼 · 发布于 2024-05-21 02:12:29

正如注释部分所提到的,不管出于什么原因,这段代码只是没有正确地存储我的列表,但是当提示打印列表时,它被存储并且脚本运行正常。以下是完成的脚本:

import numpy as np
from collections import defaultdict

lb = np.genfromtxt('file2009.csv',dtype = None, delimiter = ',', names = True)
vms = np.genfromtxt('2009final.csv',dtype = None, delimiter = ',', names = True)

id_list = list(set(lb['ID']))

matchList = defaultdict(list)

for e in id_list:
    log = lb[lb['ID']==e]
    vms_list = vms[ np.logical_and(vms['date']==log[0]['date'] , vms['code']==log[0]['code'] )]
    print vms_list 
    vms_id_list = list(set(vms_list['ID']))
    for row in log:
        if len(vms_id_list) == 1:
            break
        else:
            vmsIDs = vms[vms['ID'] == vms_id_list[0]]
            vms_ids = str(vms_id_list)
            for every in vms:
                if str(every['ID']) in vms_ids:
                    vmsIDs = np.hstack([vmsIDs, every])
            vms_id_list = []
            vms_list = vmsIDs[ np.logical_and(vmsIDs['date']==row['date'] , vmsIDs['code']==row['code'] )]
            vms_id_list = list(set(vms_list['ID']))
    matchList[e].append(vms_id_list)

相关问题 更多 >