Python:如何创建循环以逐行比较熊猫中的值

时间:2018-10-15 08:15:02

标签: python pandas loops csv

我仍然是python的新用户,并尝试使用python制作最长的普通子序列。但是我想知道如何创建一个循环来首先逐行比较值。这是数据示例。

row col1  col2  col3  col4   col5  col6 
0   'A'   '11'  'B'   '122'   ''    '' 
1   'A'   '11'  'B'   '122'  '121'  '' 
2   'A'   '12'  'B'   '123'  '124'  '' 
3   'A'   '13'  'B'   '123'  '124'  '456' 

这是我要适应的LCS的示例代码。

import pandas as pd

df_hier = pd.read_csv("sample.csv")

row1 = df_hier.iloc[1:2, 2:20]
row2 = df_hier.iloc[3:4, 2:20]

    def lcs(s1, s2):
        matrix = [["" for x in range(len(s2))] for x in range(len(s1))]
        for i in range(len(s1)):
            for j in range(len(s2)):
                if s1[i] == s2[j]:
                    if i == 0 or j == 0:
                        matrix[i][j] = s1[i]
                    else:
                        matrix[i][j] = matrix[i-1][j-1] + s1[i]
                else:
                    matrix[i][j] = max(matrix[i-1][j], matrix[i][j-1], key=len)

        cs = matrix[-1][-1]

        return len(cs), cs

    print(lcs(row1, row2))  

假设

the row1 will use 
'A'   '11'  'B'   '122'  '121'  '' 
and row2
'A'   '12'  'B'   '123'  '124'  ''

预期结果将是

(7,'A1B1212')

1 个答案:

答案 0 :(得分:0)

没有那么困难。您只需要使用zip形成值对,然后应用您已经拥有的lcs函数即可。试试这个功能:

def lcs_rows(r1, r2):
    pairs = zip(r1.values.flatten(), r2.values.flatten())
    lcss = [lcs(*x) if len(x[0]) and len(x[1]) else (0, '') for x in pairs]
    nums = [x[0] for x in lcss]
    strs = [x[1] for x in lcss]
    return (sum(nums), ''.join(strs))