Python - CSV - 如何仅从 CSV 的 header 中删除某些字符？

Question

我在一个文件夹中有 CSV，我想从特殊字符中清除标题（并且只有标题），然后将更新的 CSV 保存在一个新文件夹中。

我遇到的问题是，特殊字符不仅从标题中删除，还从下面的其他行中删除。

我的代码如下所示：

from pathlib import Path
import pandas as pd
import os

parent_dir = input("Enter CSV directory path:")
newdir = "Processed"
directory = os.path.join(parent_dir, newdir)
os.mkdir(directory)
csv_files = [f for f in Path(parent_dir).glob('*.csv')]

for csv in csv_files:
    data = pd.read_csv(csv, encoding = 'ISO-8859-1', engine='python', delimiter = ',')
    data.columns = data.columns.str.replace('[",@]','')
    data.to_csv(parent_dir + "/Processed/" + csv.name, index=False)

有关纠正此问题的任何建议？

Answer 1

尝试这个

df.columns = df.columns.str.replace(r"[^a-zA-Z\d\_]+", "")

它将删除除属于英文字母、空格和制表符的字母之外的所有字符

Answer 2

像这样一一替换字符

import pandas as pd

# generate sample df
foo = pd.DataFrame(columns=['a@', 'b[', 'c]'])

# select characters to drop
chars_to_drop = ['@', '[', ']']

for char in chars_to_drop:
   foo.columns = foo.columns.str.replace(char, '')

print(foo.columns)
>>> Index(['a', 'b', 'c'], dtype='object')

Answer 3

您可以尝试以下方法：

要从 header 中删除：

data.columns = data.columns.str.replace('[^\w\s]|_', '', regex=True)

从所有行中删除

data = data.replace(r'[^\w\s]|_', '', regex=True)