Подсчет значений в кадре данных на основе записи - PullRequest
1 голос
/ 08 января 2020

У меня есть кадр данных в виде:

category | value |
cat a    |x      |
cat a    |x      |
cat a    |y      |
cat b    |w      |
cat b    |z      |

Я хотел бы иметь возможность возвращать что-то вроде (показывая уникальные значения и частоту)

category | freq of most common value |most common value |
cat a       2                              x
cat b       1                              w #(it doesnt matter if here is an w or z)

Ответы [ 3 ]

2 голосов
/ 08 января 2020

Используйте Series.value_counts с Series.head для групп в лямбда-функции:

df = (df.groupby('category', sort=False)['value']
        .apply(lambda x: x.value_counts().head(1))
        .reset_index()
        .rename(columns={'level_1':'most_common_value','value':'freq of most common value'}))
print (df)
  category most_common_value  freq of most common value
0    cat a                 x                          2
1    cat b                 w                          1
2 голосов
/ 08 января 2020

Один из подходов состоит в том, чтобы groupby в оба столбца взять size, отсортировать значения и выбрать значение с более высокой частотой:

(df.groupby(['category', 'value'])
   .value.size()
   .sort_values()
   .groupby(level=0)
   .tail(1))

category  value
cat b      z        1
cat a      x        2
Name: value, dtype: int64
1 голос
/ 08 января 2020

Вот решение, использующее crosstab:

m = pd.crosstab(df['category'],df['value'])
m = m.max(1).to_frame('freq of most common value').assign(most_common_value=m.idxmax(1))

print(m)

          freq of most common value most_common_value
category                                             
cat a                             2                 x
cat b                             1                 w
...