Обчисліть відсоток аналогічних значень у фреймі даних панди


14

У мене є один фрейм даних dfз двома стовпцями: Сценарій (з текстом) і Speaker

Script  Speaker
aze     Speaker 1 
art     Speaker 2
ghb     Speaker 3
jka     Speaker 1
tyc     Speaker 1
avv     Speaker 2 
bhj     Speaker 1

У мене є такий список: L = ['a','b','c']

З наступним кодом:

df = (df.set_index('Speaker')['Script'].str.findall('|'.join(L))
        .str.join('|')
        .str.get_dummies()
        .sum(level=0))
print (df)

Я отримую цей кадр даних df2:

Speaker     a    b    c
Speaker 1   2    1    1
Speaker 2   2    0    0
Speaker 3   0    1    0

Який рядок я можу додати до свого коду, щоб отримати для кожного рядка мого кадру даних df2відсоткове значення всіх рядків, вимовлених диктором, щоб мати такий фрейм даних df3:

Speaker     a    b    c
Speaker 1   50%  25%   25%
Speaker 2  100%    0   0
Speaker 3   0   100%   0

Відповіді:


8

Ви можете розділити sumна першу вісь, а потім наділити рядок і додати %:

out = (df.set_index('Speaker')['Script'].str.findall('|'.join(L))
         .str.join('|')
         .str.get_dummies()
         .sum(level=0))

(out/out.sum(0)[:,None]).mul(100).astype(int).astype(str).add('%')

            a     b    c
Speaker                  
Speaker1   50%   25%  25%
Speaker2  100%    0%   0%
Speaker3    0%  100%   0%

5

Починаючи з початкового фрейму даних, якщо ви хочете% і не згрупувати суму манекерів, ви можете змінити весь сценарій, як показано нижче:

m = df.set_index('Speaker')['Script'].str.findall('|'.join(L)) #creates a list of matches
m = m.explode().reset_index() #explode to a series 
final = pd.crosstab(m['Speaker'],m['Script'],normalize='index').mul(100) # percentage pivot

Script         a      b     c
Speaker                      
Speaker 1   50.0   25.0  25.0
Speaker 2  100.0    0.0   0.0
Speaker 3    0.0  100.0   0.0

Якщо ви не хочете відсотку, просто використовуйте:

pd.crosstab(m['Speaker'],m['Script'])

Script     a  b  c
Speaker           
Speaker 1  2  1  1
Speaker 2  2  0  0
Speaker 3  0  1  0

Примітка. Для використання в цьому варіанті панди 0,25+


3
(df.set_index('Speaker')['Script'].str.extractall(f'({"|".join(L)})')
   .groupby('Speaker')[0].value_counts(normalize=True)
   .unstack(fill_value=0)
)

Вихід:

0            a     b     c
Speaker                   
Speaker 1  0.5  0.25  0.25
Speaker 2  1.0  0.00  0.00
Speaker 3  0.0  1.00  0.00

2

Надавши приклад, ви можете спробувати наступний рядок коду:

df = (df/df.sum(axis=1)[:, None]).mul(100).astype(int)

З наданими вами даними:

import pandas as pd
import numpy as np
data = {'a':[2,2,0],'b':[1,0,1],'c':[1,0,0]}
df = pd.DataFrame(data)
df = (df/df.sum(axis=1)[:, None]).mul(100).astype(int)
print(df)

Вихід:

     a   b   c
0   50  25  25
1  100   0   0
2    0 100   0

Або, якщо ви хочете додати символ "%":

df = (df / df.sum(axis=1)[:, None]).mul(100).astype(int).astype(str) + '%'

Вихід:

      a     b    c
0   50%   25%  25%
1  100%    0%   0%
2    0%  100%   0%
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.