如何根据列前缀将列的值完全替换为NA？

import pandas as pd

def replace_values_with_na(df, prefix):
    columns = df.columns
    for column in columns:
        if column.startswith(prefix):
            df[column] = 'NA'
    return df

# 示例数据集
data = {'ID': [1, 2, 3],
        'A_Value': [10, 20, 30],
        'B_Value': [40, 50, 60],
        'C_Value': [70, 80, 90]}

df = pd.DataFrame(data)

# 调用函数替换列值为NA
prefix = 'A_'
df = replace_values_with_na(df, prefix)

print(df)

   ID A_Value  B_Value  C_Value
0   1      NA       40       70
1   2      NA       50       80
2   3      NA       60       90