如何在 for 循環中的數據框中插入新列？

Question

超級新手 Python/Pandas 問題。

我正在嘗試讀取包含多個工作表的 Excel 工作簿文件夾，將列標題提取為列表，並將每個列表添加為 pandas DataFrame 中的新列。 這是我到目前為止的代碼：

import pandas as pd
import numpy as np

filepath = '/content/data.xlsx'

workbook = pd.read_excel(filepath, None, nrows=0)

variables_frame = []

for sheet_name, sheet in workbook.items():
  variables = sheet.columns
  variables_list = list(variables)
  variables_frame = pd.DataFrame.insert(sheet+1, sheet_name, [variables_list])
  print(variables_frame)

但是，當我嘗試運行它時，出現錯誤“TypeError: insert() missing 1 required positional argument: 'value'”。 任何想法為什么？

此外，如果這不是解決此問題的正確方法，我將不勝感激任何更一般的反饋。 謝謝！

Answer 1

問題是您直接從類中調用insert函數

class A:
    def insert(self, x):
        print(f'insert {x}')

# Right
A().insert(3)

# Wrong
A.insert(3)

你可能想要

variables_frame = pd.DataFrame()
for sheet_name, sheet in workbook.items():
  variables = sheet.columns
  variables_list = list(variables)
  variables_frame.insert(sheet+1, sheet_name, [variables_list])
  print(variables_frame)