Блог

Я сравниваю исходный и целевой набор данных(9000 строк,14 столбцов) и выделяю разницу, используя приведенный ниже код. здесь с помощью apply метода я вызываю функцию. но проблема в том, что выделение различий работает до некоторых строк, после чего строки не выделяют различия. Не уверен, в чем заключается ограничение, или я что-то здесь упускаю?

Примечание: Я не уверен,как прикрепить файл набора данных к этому вопросу, но я использую образец csv с 9000 строками и 14 столбцами отсюда

пример здесь, в первой записи, подчеркивает разницу,

но после идентификатора 4144 он не выделяет его.

ниже приведен мой полный код-

 import pandas as pd
import numpy as np
from IPython.display import display, HTML

def join_datasets(source, target):
    source_df = pd.read_csv(source)
    target_df = pd.read_csv(target)
    combined_df = pd.concat([source_df.set_index('id'), target_df.set_index('id')], 
                            axis='columns', keys=['Source', 'Target'])
    combined_df = combined_df.swaplevel(axis='columns')[source_df.columns[1:]]
    return combined_df

def highlight_diff(data, color='yellow'):
    attr = 'background-color: {}'.format(color)
    other = data.xs('Source', axis='columns', level=-1)
    return pd.DataFrame(np.where(data.ne(other, level=0), attr, ''),
                        index=data.index, columns=data.columns)

custom_style = [
    {
        'selector':'caption',
        'props':[('font-weight', 'bold'),('margin-bottom','25px'),('font-size','25px')]
    },
    {  
        'selector': '',
        'props': [('border-collapse', 'collapse'),('margin', '100px auto')]
    },
    {  
        'selector': 'table',
        'props': [('border-collapse', 'collapse'),('margin', '100px auto')]
    },
    {  
        'selector': 'tbody',
        'props': [('border', '1px solid #A2DBFA')]
    },
    {
        'selector': 'td',
        'props': [('border', '1px solid #A2DBFA'),('padding','1em'),('border-bottom','2px solid #A2DBFA')]
    },
    {
        'selector': 'th',
        'props': [('border', '1px solid #A2DBFA'),('padding','1em'),('background-color','#39A2DB'),('border-bottom','2px solid #A2DBFA')]
    }
]

df = join_datasets('source.csv', 'target.csv')
df = df.style.set_table_styles(custom_style, overwrite=True).set_caption("Test Case 1").apply(highlight_diff, axis=None)
df