如何替换 Pandas Dataframe 的所有列标题中的字符串？

Question

我在 Azure 上的 Databricks 中使用 Python。我尝试使用 str.replace 但我收到一条错误消息，指出列名不是字符串类型。原始文件作为 parquet 文件上传；我不确定这是否与字符串错误有关。关于如何删除前两个字符的任何想法？

我附上了一张表格的图片以供参考和字符串错误的图片。

df.columns = df.columns.str.replace('T$', '')

Answer 1

这将起作用：

old_names = df.columns
new_names = [name.replace('T$', '') for name in old_names]
df.rename(columns={old_name: new_name for old_name, new_name in zip(old_names, new_names)}, inplace=True)

Answer 2

$ 在 str.replace 中有特殊的含义。最好使用反斜杠将其转义，如下所示。

df.columns = df.columns.str.replace('T\$', '')