我有两个列表对,每个包括标识符的列表和值的列表,其中a和b不具有相同的长度。 例如:
a_id = [1, 2, 4, 5, 9, 12, 13]
a_val = [13., 32., 5., 9., 32., 4., 8.]
b_id = [1, 3, 4, 6, 9]
b_val = [12., 27., 1., 3., 19.]
现在,我需要知道至极值对应相同的ID,我只需要那些在a和b的值。 对于这个例子,我想获得共同的ID和相应值的列表:
common_id = [1, 4, 9]
common_a_val = [13., 5., 32.]
common_b_val = [12., 1., 19.]
什么是实现这一最佳/最快的方法是什么?
>>> common_id = [i for i in a_id if i in b_id]
>>> common_id
[1, 4, 9]
>>> common_a_val = [a_val[a_id.index(i)] for i in common_id]
>>> common_a_val
[13.0, 5.0, 32.0]
>>> common_b_val = [b_val[b_id.index(i)] for i in common_id]
>>> common_b_val
[12.0, 1.0, 19.0]
>>> a_d = dict(zip(a_id,a_val))
>>> b_d = dict(zip(b_id,b_val))
>>> common_ids = a_d.viewkeys() & b_d.viewkeys()
set([1, 4, 9])
>>> common_a_val = [a_d[key] for key in common_ids]
[13.0, 5.0, 32.0]
>>> common_b_val = [b_d[key] for key in common_ids]
[12.0, 1.0, 19.0]
def common_elements(list1, list2):
return [element for element in list1 if element in list2]
a_id = [1, 2, 4, 5, 9, 12, 13]
a_val = [13., 32., 5., 9., 32., 4., 8.]
b_id = [1, 3, 4, 6, 9]
b_val = [12., 27., 1., 3., 19.]
common_a_val=[];common_b_val=[]
common_id=common_elements(a_id,b_id)
for i in common_id:
common_a_val.append(a_val[a_id.index(i)])
common_b_val.append(b_val[b_id.index(i)])
print common_id,common_a_val,common_b_val
它的输出是:
[1, 4, 9] [13.0, 5.0, 32.0] [12.0, 1.0, 19.0]