使用apply()突出Pandas DataFrame的特定列

  • Post category:Python

使用apply()方法实现对Pandas DataFrame特定列突出的方法如下:

步骤1:在Pandas中读入数据集,并对数据进行必要的处理。这里以读入Titanic数据集为例:

import pandas as pd

data = pd.read_csv('titanic.csv')
df = pd.DataFrame(data)

步骤2:使用apply()方法,定义函数并将该函数应用于想要突出的特定列。这里以对Titanic数据集中的Fare列进行突出为例:

def highlight_fare(val):
    if val >= 80:
        return 'background-color: yellow'
    else:
        return 'background-color: white'

df.style.applymap(highlight_fare, subset=['Fare'])

在这个例子中,我们定义了名为highlight_fare()的函数,如果Val的值大于等于80,那么函数将返回黄色背景,否则返回白色背景。我们用applymap()方法将其应用到Fare列上。

步骤3:运行以上代码,Pandas会为符合条件的单元格设置黄色背景。

下面是完整的代码示例:

import pandas as pd

data = pd.read_csv('titanic.csv')
df = pd.DataFrame(data)

def highlight_fare(val):
    if val >= 80:
        return 'background-color: yellow'
    else:
        return 'background-color: white'

df.style.applymap(highlight_fare, subset=['Fare'])

运行以上代码,你会看到Titanic数据集中Fare列的值大于等于80的单元格被突出显示。