Ошибка dataFrame.drop() в Python PywedgeCharts

Я пытался использовать Pywedge_Charts с файлом csv (простой файл csv с 3 столбцами (‘Штат’, ‘Налоги’, ‘Годовые_доходы’)), кодируя следующее:

import pandas as pd
import pywedge as pw

tax = pd.read_csv('C:/Users/XXXX/OneDrive/Documents/Python/Taxes.csv')

viz = pw.Pywedge_Charts(tax, c=None, y='State')

но я получаю следующее сообщение об ошибке:

TypeError                                 Traceback (most recent call last)
`your text`Cell In[28], line 1
----> 1 viz = pw.Pywedge_Charts(tax, c=None, y='Taxes')

File ~\anaconda3\Lib\site-packages\pywedge\pywedge.py:27, in Pywedge_Charts.__init__(self, train, c, y, manual)
     25 self.c = c
     26 self.y = y
---> 27 self.X = self.train.drop(self.y,1)
     28 self.manual = manual

TypeError: DataFrame.drop() takes from 1 to 2 positional arguments but 3 were given

Пробовал разные варианты, но не смог решить проблему. Есть ли идеи, как исправить эту ошибку?

Я новичок в Python, но это кажется базовым кодом, который не должен вызывать проблем. Пробовал присвоить другой столбец y, изменить файл csv, но я все равно сталкиваюсь с этой проблемой.

Заранее спасибо

С учетом того, что последняя версия pywedge была выпущена 20 января 2021 года, она естественно полагается на старую версию df.drop:

• df.drop <= 1.4.x
• df.drop >= 1.5.x

Старая сигнатура:

DataFrame.drop(labels=None, axis=0, index=None, columns=None, 
               level=None, inplace=False, errors="raise")

Новая сигнатура:

DataFrame.drop(labels=None, *, axis=0, index=None, columns=None, 
               level=None, inplace=False, errors="raise")

Обратите внимание на * во второй позиции в новой сигнатуре. Этот символ указывает, что все аргументы, которые следуют, должны передаваться как именованные аргументы. Смотрите эту документацию. В старой версии этот * отсутствует, поэтому вы могли написать:

df.drop('label', 1)

# это значит: df.drop(labels="label", axis=1)

Как видно из сообщения об ошибке, это реализация в pywedge.py:

self.train.drop(self.y,1)

Новая версия больше не распознает 1 как позиционный аргумент для axis. Следовательно, вам нужно написать:

df.drop('label', axis=1)

Решение 1

Понизить вашу версию pandas до 1.4.4 (или ниже).

Решение 2

Найдите установленный pywedge.py и исправьте все выражения .drop(...). Я насчитал четыре, начиная отсюда. То есть,

self.X = self.train.drop(self.y,1)

должно стать:

self.X = self.train.drop(self.y, axis=1)

Решит ли одно из этих решений все проблемы, – загадка. requirements.txt не уточняет версии для большинства зависимостей. Следовательно, оба решения лучше всего сочетать с маленькой молитвой.

Пример

Функция drop в pandas используется неправильно в библиотеке, которую вы использовали. Я создам пример, чтобы показать, почему этот код неверен.

import pandas as pd
data = {'A': [59, 11, 65, 95], 'B': [63, 87, 22, 87], 
        'C': [49, 51, 9, 99], 'D': [4, 39, 77, 65]}
df = pd.DataFrame(data)

df

    A   B   C   D
0  59  63  49   4
1  11  87  51  39
2  65  22   9  77
3  95  87  99  65

Попробуйте удалить столбец ‘A’ из df, и вы получите ту же ошибку, если упустите параметр axis.

df.drop('A', 1)

ошибка

TypeError: DataFrame.drop() takes from 1 to 2 positional arguments but 3 were given

Мы можем удалить столбец ‘A’ из df без ошибки, если не упустим параметр axis.

df.drop('A', axis=1)

Похоже, что библиотека pywedge, которую вы использовали, основана на более старой версии Pandas, и поэтому она упускает параметр axis в функции drop. Я думаю, что правильное понижение версии Pandas (возможно, 1.4.4, основываясь на ответе @ouroboros1) должно это исправить.