我只是在使用 Python 2.7 学习 Python。我有一个包含两列的 csv 文件。这些列是:
Coll_id:条目可能是单个收集器,也可能是组
Participant_Col_id:如果 Coll_id 是单个收集器,则该值将为 null。如果 Coll_id 是一个组,则该组中的每个参与者将有一行。
这里有一个示例:
Coll_id,Participant_Coll_id<br>
ARA,ARG
ARA,RAT
ARG,NULL
BRSAR,SGMB
BRSAR,SANTM
BRSAR,CRSR
BRSAR,RAT
CRSR,NULL
DBY,NULL
HZIE,NULL
RAT,NULL
SANTM,NULL
SGMB,NULL
ARG,NULL
DRS,CRSR
DRS,RAT
DRS,ARG
对于每个收集器(coll_id),我正在尝试创建他们收集的所有其他收集器的列表。我试图将代码放在一起来执行以下操作,现在已经很接近了:
#This is giving me a dictionary with each COLL_ID having a list of PARTICIPANT_COLL_IDs
with open('colls_mv1.csv', 'r') as f:
reader = csv.DictReader(f, ['COLL_ID', 'PARTICIPANT_COLL_ID'])
data1 = defaultdict(list)
for line in reader:
data1[line['COLL_ID']].append(line['PARTICIPANT_COLL_ID'])
#And this is giving me a dictionary with each PARTICIPANT_COLL_ID having a list of COLL_IDs
with open('colls_mv1.csv', 'r') as f:
reader = csv.DictReader(f, ['COLL_ID', 'PARTICIPANT_COLL_ID'])
data2 = defaultdict(list)
for line in reader:
if line['PARTICIPANT_COLL_ID'] != 'NULL':
data2[line['PARTICIPANT_COLL_ID']].append(line['COLL_ID'])
dict3 = {k: [data1[i] for i in v] for k, v in data2.items()}
print dict3
我得到以下输出:
{'SGMB': [['SGMB', 'SANTM', 'CRSR', 'RAT']], 'CRSR': [['SGMB', 'SANTM', 'CRSR', 'RAT'], ['CRSR', 'RAT', 'ARG']], 'RAT': [['ARG', 'RAT'], ['SGMB', 'SANTM', 'CRSR', 'RAT'], ['CRSR', 'RAT', 'ARG']], 'PARTICIPANT_COLL_ID': [['PARTICIPANT_COLL_ID']], 'ARG': [['ARG', 'RAT'], ['CRSR', 'RAT', 'ARG']], 'SANTM': [['SGMB', 'SANTM', 'CRSR', 'RAT']]}
我想要的是将每个键的值列表合并在一起,删除重复项并从值列表中删除键:
{'SGMB': ['SANTM', 'CRSR', 'RAT'], 'CRSR': ['SGMB', 'SANTM', 'RAT', 'ARG'], 'RAT': ['ARG', 'SGMB', 'SANTM', 'CRSR'], 'PARTICIPANT_COLL_ID': [['PARTICIPANT_COLL_ID']], 'ARG': ['RAT', 'CRSR'], 'SANTM': ['SGMB', 'CRSR', 'RAT']}