将两个列表列表与 dataframe 列进行比较 python

Question

我想将两个列表列表与 dataframe 列进行比较。
list1=[[r2,r4,r6],[r6,r7]]
list2=[[p4,p5,p8],[p86,p21,p0,p94]]

数据集：

摆脱	进程号	价值
r2	p0	香蕉
r2	p4	巧克力
r4	p89	苹果
r6	p5	牛奶
r7	p0	面包

Output：

[[chocolate,milk],[bread]]

由于r2和p4出现在list1[0] 、 list2[0]和数据集中的同一行中，因此必须存储chocolate 。 类似地， r6和p5出现在两个列表中的相同 position 和数据集中的同一行中，必须存储milk 。

Answer 1

您可以按如下方式进行：

from itertools import product

df = pd.DataFrame({'rid': {0: 'r2', 1: 'r2', 2: 'r4', 3: 'r6', 4: 'r7'},
 'pid': {0: 'p0', 1: 'p4', 2: 'p89', 3: 'p5', 4: 'p0'},
 'value': {0: 'banana', 1: 'chocolate', 2: 'apple', 3: 'milk', 4: 'bread'}})
list1 = [['r2','r4','r6'],['r6','r7']]
list2 = [['p4','p5','p8'],['p86','p21','p0','p94']]

# Generate all possible associations.
associations = (product(l1, l2) for l1, l2 in zip(list1, list2))

# Index for speed and convenience of the lookup.
df = df.set_index(['rid', 'pid']).sort_index()

output = [[df.loc[assoc, 'value'] for assoc in assoc_list if assoc in df.index] 
          for assoc_list in associations]

print(output)

[['chocolate', 'milk'], ['bread']]

Answer 2

回答

result = []
for l1, l2 in zip(list1, list2):
    res = df.loc[df["rid"].isin(l1) & df["pid"].isin(l2)]["value"].tolist()
    result.append(res)

[['chocolate', 'milk'], ['bread']]

解释

• zip将合并两个列表，相当于

for i in range(len(list1)):
    l1 = list1[i]
    l2 = list2[i]

• df["rid"].isin(l1) & df["pid"].isin(l2)将条件与and operator &组合

署名

• list1 和 list2 的长度必须相等，否则zip将忽略较长列表的 rest 元素。