将功能应用于数据框中的特定行

Question

我有一个要清理特定行的数据框，在本例中为第一行。 我编写了一个函数，如果正则表达式匹配，我相信该函数将返回字符串。

def clean_cells(string):
    if '201' in string:
        return re.findall('201[0-9]', string)[0]
    else:
        return string

我想将此功能应用于数据框的第一行，然后将第一行替换为清理后的版本，然后处理原始数据框的其余部分。

我试过了：

df = df.iloc[[0]].apply(clean_cells)

Answer 1

按位置选择第一行，并使用:选择所有列，然后分配：

df.iloc[0, :] = df.iloc[0, :].apply(clean_cells)

另一个解决方案：

df.iloc[0] = df.iloc[0].apply(clean_cells)

Answer 2

仅替换选定的行，而不是整个DataFrame：

df.iloc[[0]] = df.iloc[[0]].apply(clean_cells)