我用日期和位置代码匹配两个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中一行的条件相匹配。但是它仍然不起作用,列表是空的。有什么锁我没看见吗?有没有别的方法来做我想做的事?你知道吗
正如注释部分所提到的,不管出于什么原因,这段代码只是没有正确地存储我的列表,但是当提示打印列表时,它被存储并且脚本运行正常。以下是完成的脚本:
相关问题 更多 >
编程相关推荐