0

我通过日期和位置代码匹配来自两个 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 (lb) 中的记录匹配的所有记录的第二个 csv (vms) 创建一个子集数组。这是我正在使用的代码：

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_list 为空，稍后会在该行生成错误vmsIDs = vms[vms['ID'] == vms_id_list[0]]。我真的很困惑，因为创建了列表，即使我知道有很多符合条件的记录，它也只是完全空的。
如果我只是将这行代码复制出循环并独立运行它，它工作正常，所以问题在于它在循环中执行。我不知道为什么这不起作用，我什至尝试将语言从 numpy logical_and 更改为：

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 中获取与 lb 中的一行的条件匹配的 ID 列表。但它仍然不起作用，列表为空。是否有某种我没有看到的锁？有没有不同的方法来做我想做的事情？

4

1 回答 1

0

如评论部分所述，无论出于何种原因，这段代码都没有正确存储我的列表，但是当提示打印列表时，它被存储并且脚本正确运行。这是完成的脚本：

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)
于 2013-08-28T16:48:14.263 回答