下面有一个元组列表形式的模板,我将使用DataFrameJoin对其进行实例化
rule = [('#1', 'X', 'Y'), ('#2', 'X', 'Z'), ('#3', 'Z', 'Y')]
我还有一个作为字典的模板的每个组件的实例
rComp_substitution =
{('#1', 'X', 'Y'): pred subj obj
0 nationality BART USA,
('#2', 'X', 'Z'): pred subj obj
0 placeOfBirth BART NEWYORK
1 hasFather BART HOMMER,
('#3', 'Z', 'Y'): pred subj obj
0 locatedIn NEWYORK USA
1 nationality HOMMER USA }
每个组件对应的实例是一个dataframe,有三列。对于('#1', 'X', 'Y')
,#1
对应于pred
,X
对应于subj
,而Y
对应于obj
例如,首先实例化('#1',X',Y'),('#2',X',Z')
我们可以检查('#1',X',Y')和('#2',X',Z')的公共变量
并用一个键连接每个数据帧的公共变量X(subc),以获得('#1',X',Y'),('#2',X',Z')的实例
下面是我的代码
depth = 0
# step1 check common variable
current_subj = rule[depth][1] #['X']
current_obj = rule[depth][2] #['Y']
next_subj = rule[depth+1][1] #['X']
next_obj = rule[depth+1][2] #['Z']
if current_subj == next_subj or current_subj == next_obj:
comVar = current_subj
elif current_obj == next_subj or current_obj == next_obj:
comVar = current_obj
# step2 Create currnt_rComp with common variable for joining dataframes
current_rComp = rComp_substitution[rule[depth]]
unified_rComp = []
for col in current_rComp.itertuples(index=False):
if comVar == current_subj:
unified_rComp.append([col.subj, [list(col)]])
elif comVar == current_obj:
unified_rComp.append([col.obj, [list(col)]])
current_rComp = pd.DataFrame(unified_rComp, columns=['comVar', 'triples'])
# step3 Create next_rComp with common variable for joining dataframes
next_rComp = rComp_substitution[rule[depth+1]]
unified_rComp = []
for col in next_rComp.itertuples(index=False):
if comVar == next_subj:
unified_rComp.append([col.subj, [list(col)]])
elif comVar == next_obj:
unified_rComp.append([col.obj, [list(col)]])
next_rComp = pd.DataFrame(unified_rComp, columns=['comVar', 'triples'])
# step4 Join currnt_rComp and next_rComp with common variable as key
partial_proof_path = pd.merge(current_rComp, next_rComp, how='inner', on='comVar')
print(partial_proof_path)
此代码输出为
comVar triples_x triples_y
0 BART [[nationality, BART, USA]] [[placeOfBirth, BART, NEWYORK]]
1 BART [[nationality, BART, USA]] [[hasFather, BART, HOMMER]]
我认为这段代码太长了。对于更简单的代码,有没有同样的方法
输入数据:
主要功能:
用法:
相关问题 更多 >
编程相关推荐