python - 使用 Pandas/Python 在给定相应数据的情况下填充 excel 中的列

Question

我正在尝试编写一个 pandas/python 脚本，在 jupyter notebooks 中执行以下操作 see excel data for example

我需要为每一行数据搜索 C 列，并查看 E 列​​中该行对应的数字。然后我希望它在 G 列中查找相同的数字，并将它从 E 中获得的相应值放入进入第 I 列。

如果 C 列中的某个值在 Col E 中有不同的对应值的多个实例，请标记这些 C 列值，以便我查看。

给定 C 列包含 111 并且在 E 列中有代码“a”，代码“a”将放置在 I 列中 G 列编号为 111 的任何位置。

如果他们没有相同的数字，用红色突出显示 C 列中的那些值。

我不知道如何编写代码。如果有人能告诉我，将不胜感激。谢谢

Answer 1

这就是你想要的..

dct = {'C':[111,222,333,111,444],'E':['a','b','c','d','e'],'G':[111,123,333,111,444]}

df = pd.DataFrame(dct)

highlight = []
vals = []
for i in range(len(df)):
    if df['C'][i] == df['G'][i]:
        highlight.append(False)
        vals.append(df['E'][i])
    else:
        highlight.append(True)
        vals.append(None)

df['I'] = vals

def highlight_cells(x):
    c1 = 'background-color: red'
    c2 = '' 

    df1 =  pd.DataFrame(c2, index=df.index, columns=df.columns)
    #modify values of df1 column by boolean highlight

    df1.loc[highlight, 'C'] = c1 #new styled dataframe

    return df1

df.style.apply(highlight_cells, axis=None).to_excel('styled.xlsx', engine='openpyxl')

首先准备高亮列表( bool 值)，即我们正在标记 colC 的哪些行需要高亮显示。现在我们在函数 highlight_cells 中使用这个高亮列表，它创建新的屏蔽数据帧并使用 df.style.apply() 将其应用于数据帧 df。

输出: