У меня есть некоторые школьные данные:
data = {'name': ['school a', 'school b', 'school c', 'school d', 'school e', 'school f'],
'type': ['a', 'a', 'b', 'b', 'a', 'b'],
'location': ['county a', 'county a', 'county b', 'county b', 'county b', 'county a'],
'avg_score': [9, 7, 5, 7, 6, 8]
}
df = pd.DataFrame(data)
Out:
name type location avg_score
0 school a a county a 9
1 school b a county a 7
2 school c b county b 5
3 school d b county b 7
4 school e a county b 6
5 school f b county a 8
Я хотел бы включить сравнение школьных баллов со средним типом школы по местоположению.
Я могу сделать это сgroupby: df.groupby(['type', 'location']).mean().round(2)
Out:
avg_score
type location
a county a 8
county b 6
b county a 8
county b 6
Однако я хотел бы получить дополнительный столбец со средним значением для этого типа школы для местоположения вместо сгруппированной таблицы.
Как я могу получить сравнение: вот так:
name type location avg_score compare_score
0 school a a county a 9 8
1 school b a county a 7 5
2 school c b county b 5 7
3 school d b county b 7 7
4 school e a county b 6 3
5 school f b county a 8 7
Я нашел этот вопрос
Среднее значение Python Pandas на основе условия в новом столбце
и попытался применить некоторые из возможных решений моей проблемы:
for atype, alocation in df.groupby('type'):
df.loc[df.type == type, 'compare'] = (df.where(df['type' == atype]).where(df['location' == alocation]).mean()).avg_score.round(2)```
Вызывает эту ошибку:
---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
~/.local/share/virtualenvs/schule-jwiURUl3/lib/python3.6/site-packages/pandas/core/indexes/base.py in get_loc(self, key, method, tolerance)
2656 try:
-> 2657 return self._engine.get_loc(key)
2658 except KeyError:
pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
KeyError: False
During handling of the above exception, another exception occurred:
KeyError Traceback (most recent call last)
<ipython-input-27-4b3cf2b7aaf6> in <module>
1 for atype, alocation in df.groupby('type'):
----> 2 df.loc[df.type == type, 'compare'] = (df.where(df['type' == atype]).where(df['location' == alocation]).mean()).avg_score.round(2)
~/.local/share/virtualenvs/schule-jwiURUl3/lib/python3.6/site-packages/pandas/core/frame.py in __getitem__(self, key)
2925 if self.columns.nlevels > 1:
2926 return self._getitem_multilevel(key)
-> 2927 indexer = self.columns.get_loc(key)
2928 if is_integer(indexer):
2929 indexer = [indexer]
~/.local/share/virtualenvs/schule-jwiURUl3/lib/python3.6/site-packages/pandas/core/indexes/base.py in get_loc(self, key, method, tolerance)
2657 return self._engine.get_loc(key)
2658 except KeyError:
-> 2659 return self._engine.get_loc(self._maybe_cast_indexer(key))
2660 indexer = self.get_indexer([key], method=method, tolerance=tolerance)
2661 if indexer.ndim > 1 or indexer.size > 1:
pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()
pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
KeyError: False
Возможно, это не очень хорошая попытка вообще. У Вас есть какие-то предложения? Любая подсказка высоко ценится.