向Pandas數據框添加新列時出錯

Question

我正在嘗試修改文件夾中的.csv文件。 這些文件包含2011-2016年的航班信息。

但是，在這些值中找不到年份信息。

我想通過使用包含年份的.csv文件的文件名來解決此問題。 在將其讀入pandas數據框后，我將添加一個新的“ year”列。 然后，我將修改后的文件僅以年份作為文件名導出到新的.csv。

但是，我遇到此錯誤：

ValueError：值的長度與索引的長度不匹配

下面的代碼供您參考。

import pandas as pd
import glob
import re
import os

path = r'data_caap/'                   
all_files = glob.glob(os.path.join(path, "*.csv"))


for f in all_files:
    df = pd.read_csv(f)
    year= re.findall(r'\d{4}', f)

    #Error here
    df['year']=year
    #Error here

    df.to_csv(year)

Answer 1

找到了錯誤的原因。

必須為df ['year'] = year [0]。 findall返回一個列表。 – DyZ

非常感謝@Dyz