ошибка при использовании dict для разделения фрейма данных

Question

Данные обучения выглядят следующим образом:

p,x,s,n,t,p,f,c,n,k,e,e,s,s,w,w,p,w,o,p,k,s,u
e,x,s,y,t,a,f,c,b,k,e,c,s,s,w,w,p,w,o,p,n,n,g
e,b,s,w,t,l,f,c,b,n,e,c,s,s,w,w,p,w,o,p,n,n,m
p,x,y,w,t,p,f,c,n,n,e,e,s,s,w,w,p,w,o,p,k,s,u
e,x,s,g,f,n,f,w,b,k,t,e,s,s,w,w,p,w,o,e,n,a,g
e,x,y,y,t,a,f,c,b,n,e,c,s,s,w,w,p,w,o,p,k,n,g
e,b,s,w,t,a,f,c,b,g,e,c,s,s,w,w,p,w,o,p,k,n,m    

В первом столбце указывается, является ли этот гриб съедобным (e: съедобный, p: ядовитый). И я хочу разбить эти данные на две части. часть съедобна или нет. Мой код ниже:

mushdf = pd.read_csv('agaricus-lepiota.data') #load in two data for mushroom and iris
mushdf.columns = ['edible?','cap-shape','cap-surface','cap-color','bruises?','odor',
                    'gill-attachment','gill-spacing','gill-size','gill-color',
                    'stalk-shape','stalk-root','stalk-surface-above-ring','stalk-surface-below-ring',
                    'stalk-color-above-ring','stalk-color-below-ring','veil-type','veil-color',
                    'ring-number','ring-type','spore-print-color','population','habitat']
print(mushdf)
mushdic = {key: mushdf for (key, mushdf) in mushdf.groupby('edible?')}
for key in mushdic:
    print(f'mushdic[{key}]')
    print(mushdic[key])
    print('-'*50)

Проблема в том, что когда я удаляю mushdf.columns в строке 2 - строке 6, этот код работает. Однако, когда я делаю mushdf.columns, терминал возвращает сообщение об ошибке.

Тот же метод с другим столбцом в порядке. Например, mushdic = {key: mushdf for (key, mushdf) in mushdf.groupby('bruises?')} работает правильно.

Я понятия не имею об этом.

Traceback (most recent call last):
  File "e:\Visual Studio Project\LiMing\vs2017_python\.vscode\helloworld.py", line 11, in <module>
    mushdic = {key: mushdf for (key, mushdf) in mushdf.groupby('edible?')}
  File "C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python36_64\lib\site-packages\pandas\core\generic.py", line 7894, in groupby
    **kwargs
  File "C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python36_64\lib\site-packages\pandas\core\groupby\groupby.py", line 2522, in groupby
    return klass(obj, by, **kwds)
  File "C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python36_64\lib\site-packages\pandas\core\groupby\groupby.py", line 391, in __init__
    mutated=self.mutated,
  File "C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python36_64\lib\site-packages\pandas\core\groupby\grouper.py", line 621, in _get_grouper
    raise KeyError(gpr)
KeyError: 'edible?'
The terminal process terminated with exit code: 1

Answer 1

pandas.read_csv означает, что ваша первая строка в CSV-файле является заголовком. Поскольку ваш CSV-файл не имеет заголовка, вы должны сообщить об этом во время импорта. Вы также должны передать имена столбцов здесь:

mushdf = pd.read_csv('agaricus-lepiota.data', header=None, names=[
                'edible?','cap-shape','cap-surface','cap-color','bruises?','odor',
                'gill-attachment','gill-spacing','gill-size','gill-color',
                'stalk-shape','stalk-root','stalk-surface-above-ring','stalk-surface-below-ring',
                'stalk-color-above-ring','stalk-color-below-ring','veil-type','veil-color',
                'ring-number','ring-type','spore-print-color','population','habitat'])