使用python使用文件名信息将一列批量插入到CSV中

Question

我有 169 个具有相同结构（90 列具有相同标题）和相同命名系统的 CSV 文件。

文件名截图

这些文件被命名为：

• 2019-v-1

• 2019-v-2

• 2019-v-3

• ETC。

对于每个 CSV，我想添加一列，标题为“访问”，并且该列中的值取自文件名（末尾的数字，第二个破折号之后）。

因此，例如，第一个 CSV 将有一个名为“访问”的新列，其中每一行在该列中都被赋予值“1”。

如果有 Python 解决方案，那就太棒了。我没有编码背景，这是我唯一有点熟悉的语言，但我自己似乎无法弄清楚这门语言。

任何帮助将不胜感激 - 谢谢！

Answer 1

import pandas as pdimport osdef csv_folder_input(path,folder):    path = os.path.join(path,folder)    os.chdir(path)    counter=1    for filename in os.listdir(path):        if filename.endswith(".csv"):             with open(filename, 'r') as csvfile:             counter=counter+1             df = pd.read_csv(csvfile)             df['Visits']=int(filename.split('_')[2].split('.')[0])             df.to_csv(filename)       csv_folder_input(your path name,your folder name)输入您的路径名，然后输入您的文件夹名称。我可以看到你的文件夹名称是 2019-v。在文件夹前输入适当的路径名，并确保输入适用于 MacOS 的正确路径格式。我相信它应该可以正常工作。

Answer 2

首先，您需要一个文件列表：from os import listdirfrom os.path import isfile, joinimport csv       # You'll need this for the next stepmypath = 'path/to/csv/directory'allfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]然后你想打开每个文件，添加列，然后再次保存。from os import listdirfrom os.path import isfile, joinimport csvmypath = 'path/to/csv/directory'allfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]for file in allfiles:  # This will only work if your filenames are consistent.  # We split the name into a list, ['2019', 'v', '1.csv']  #   then take the third item (index 2), and remove the  #   last 4 characters.  number_at_end = file.split("-")[2][:-4]  # We open the file...  with open(file, newline='') as csvfile:    reader = csv.reader(csvfile)  # Add the column name to the first row, and  # add the value to each row...  for i, row in enumerate(reader):    if i == 0:      row.append('Visits')    else:      row.append(number_at_end)  # and then write the file back.  with open(file, newline='') as csvfile:    writer = csv.writer(csvfile)    writer.writerows(reader)