Блог

Я столкнулся с некоторой проблемой при попытке удалить столбцы из нескольких файлов Excel в Python. Я получаю приведенную ниже ошибку, когда я пытаюсь использовать один и тот же код в одном файле, он работает, но он не работает с несколькими файлами во время цикла, и я не понимаю, в чем причина ошибки [columns ] not found in axis . Я не уверен, в чем проблема с моим кодом. Любая помощь очень ценится.

 import os
import glob
import pandas as pd
from pathlib import Path
folder = (r"C:UserskcDocumentsExtracted")


for file in Path(folder).glob('*.xlsx'):
    df = pd.read_excel(file)
    df2 = df.drop(columns=['BarrierFreeAttributes.BarrierFreeAttribute', 'ConsultationHours.ConsultationHoursTimeSpan', 'Location.Coordinates.Latitude_right', 'Location.Coordinates.Longitude_right'], axis=1)
   
    
df2.to_excel(file.with_suffix('.xlsx'),index = False)
 

ошибка:

 ---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-21-5d2d70719121> in <module>
     12     cols = cols.map(lambda x: x.replace('.','-'))
     13     df.columns = cols
---> 14     df2 = df.drop(columns=['BarrierFreeAttributes.BarrierFreeAttribute', 'ConsultationHours.ConsultationHoursTimeSpan', 'Location.Coordinates.Latitude_right', 'Location.Coordinates.Longitude_right'], axis=1)
     15 
     16 

~Anaconda3libsite-packagespandascoreframe.py in drop(self, labels, axis, index, columns, level, inplace, errors)
   4161                 weight  1.0     0.8
   4162         """
-> 4163         return super().drop(
   4164             labels=labels,
   4165             axis=axis,

~Anaconda3libsite-packagespandascoregeneric.py in drop(self, labels, axis, index, columns, level, inplace, errors)
   3885         for axis, labels in axes.items():
   3886             if labels is not None:
-> 3887                 obj = obj._drop_axis(labels, axis, level=level, errors=errors)
   3888 
   3889         if inplace:

~Anaconda3libsite-packagespandascoregeneric.py in _drop_axis(self, labels, axis, level, errors)
   3919                 new_axis = axis.drop(labels, level=level, errors=errors)
   3920             else:
-> 3921                 new_axis = axis.drop(labels, errors=errors)
   3922             result = self.reindex(**{axis_name: new_axis})
   3923 

~Anaconda3libsite-packagespandascoreindexesbase.py in drop(self, labels, errors)
   5280         if mask.any():
   5281             if errors != "ignore":
-> 5282                 raise KeyError(f"{labels[mask]} not found in axis")
   5283             indexer = indexer[~mask]
   5284         return self.delete(indexer)

KeyError: "['BarrierFreeAttributes.BarrierFreeAttribute'n 'ConsultationHours.ConsultationHoursTimeSpan'n 'Location.Coordinates.Latitude_right'n 'Location.Coordinates.Longitude_right'] not found in axis

 

Весьма вероятно, что в xlsx файле в extracted папке нет столбцов, которые вы хотите удалить. Попробуйте добавить условие фильтра для обработки остальных файлов, а также распечатать имя файла(ов) без столбцов.

 import os
import glob
import pandas as pd
from pathlib import Path
folder = (r"C:UserskcDocumentsExtracted")


for file_name in Path(folder).glob('*.xlsx'):
  df = pd.read_excel(file_name)
  drop_list = ['BarrierFreeAttributes.BarrierFreeAttribute','ConsultationHours.ConsultationHoursTimeSpan', 
'Location.Coordinates.Latitude_right', 'Location.Coordinates.Longitude_right']
    
  if all(item in drop_list for item in list(df.columns)):
       df = df.drop(columns=drop_list, axis=1)
  else:
       print(file_name)
       df.to_excel(file_name.with_suffix('.xlsx'),index = False)
 

Таким образом, вы можете обработать остальные файлы, а также распечатать подозрительные файлы, в которых могут отсутствовать столбцы, которые вы хотите удалить.
Также старайтесь не использовать file в качестве переменной, так как это зарезервированное ключевое слово в python.