Получите общее значение values_count из фрейма данных с помощью Python Pandas - PullRequest
0 голосов
/ 25 августа 2018

У меня есть датафрейм Python для панд с несколькими столбцами.Теперь я хочу скопировать все значения в один столбец, чтобы получить результат values_count со всеми включенными значениями.В конце мне нужно общее количество строк1, строк2, n.Каков наилучший способ сделать это?

index row 1    row 2   ...
0     string1  string3
1     string1  string1
2     string2  string2
...

1 Ответ

0 голосов
/ 25 августа 2018

Если производительность является проблемой, попробуйте:

from collections import Counter

Counter(df.values.ravel())
#Counter({'string1': 3, 'string2': 2, 'string3': 1})

или stack в один Series, затем используйте value_counts

df.stack().value_counts()
#string1    3
#string2    2
#string3    1
#dtype: int64

Для больших (длинных) фреймов данных снебольшое количество столбцов, циклы могут быть быстрее, чем стекирование:

s = pd.Series()
for col in df.columns:
    s = s.add(df[col].value_counts(), fill_value=0)

#string1    3.0
#string2    2.0
#string3    1.0
#dtype: float64

Кроме того, есть решение Numpy:

import numpy as np
np.unique(df.values, return_counts=True)

#(array(['string1', 'string2', 'string3'], dtype=object),
# array([3, 2, 1], dtype=int64))

df = pd.DataFrame({'row1': ['string1', 'string1', 'string2'],
                   'row2': ['string3', 'string1', 'string2']})

def vc_from_loop(df):
    s = pd.Series()
    for col in df.columns:
        s = s.add(df[col].value_counts(), fill_value=0)
    return s

Малый DataFrame

%timeit Counter(df.values.ravel())
#11.1 µs ± 56.6 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

%timeit df.stack().value_counts()
#835 µs ± 5.46 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

%timeit vc_from_loop(df)
#2.15 ms ± 34.6 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%timeit np.unique(df.values, return_counts=True)
#23.8 µs ± 241 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)

Длинный DataFrame

df = pd.concat([df]*300000, ignore_index=True)

%timeit Counter(df.values.ravel())
#124 ms ± 1.85 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

%timeit df.stack().value_counts()
#337 ms ± 3.59 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

%timeit vc_from_loop(df)
#182 ms ± 1.58 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

%timeit np.unique(df.values, return_counts=True)
#1.16 s ± 1.09 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...