时间戳

首页 > 解决方案 > 如何在 pandas 中向 .str.contains 添加多个字符串？

python - 如何在 pandas 中向 .str.contains 添加多个字符串？

问题描述

检查我提供的代码，检查 m1 和 m2 的部分，我在那里评论。我想添加多个字符串，这里有简单的解决方案吗？

我知道这不是这样做的方法，但这是我想出的唯一解决方案。如果您有其他相同想法的解决方案，请提供。

import pandas as pd
NewDataFrame = pd.DataFrame({'Group':['aaa','bbb','ccc', 'ddd', 'eee', 'fff', 'ggg', 'hhh'],
                                  'N':[1,2,3,4,5,6,7,8]})


def highlight_cells(x):
    c2 = 'background-color: #00FF00'
    c1 = 'background-color: red'
    c = 'background-color: white'

    m1 = NewDataFrame['Group'].str.contains("aaa", na=False) # here to add more strings ("bbb", "ggg") 
    m2 = NewDataFrame['Group'].str.contains("ccc", na=False) # here to add additional strings ("fff")
    #how to add more then one string, not to add new line for every string?
    
    colordata = pd.DataFrame(c, index=x.index, columns=x.columns)
    colordata.loc[m1, 'Group'] = c1
    colordata.loc[m2, 'Group'] = c2
    return colordata

end = StartingDataFrame.style.apply(highlight_cells,axis=None)

end

标签： pythonpandasdataframepandas-styles

解决方案

添加字符串，|如：

def highlight_cells(x):
    c2 = 'background-color: #00FF00'
    c1 = 'background-color: red'
    c = 'background-color: white'

    m1 = NewDataFrame['Group'].str.contains("aaa|bbb|ggg", na=False) 
    m2 = NewDataFrame['Group'].str.contains("ccc|fff", na=False) 
    #how to add more then one string, not to add new line for every string?
    
    colordata = pd.DataFrame(c, index=x.index, columns=x.columns)
    colordata.loc[m1, 'Group'] = c1
    colordata.loc[m2, 'Group'] = c2
    return colordata

end = StartingDataFrame.style.apply(highlight_cells,axis=None)

推荐阅读