Создание таблицы непредвиденных обстоятельств в Pandas - PullRequest
0 голосов
/ 06 октября 2018

Я хочу создать таблицу непредвиденных обстоятельств в Пандах.Я могу сделать это с помощью следующего кода, но мне было интересно, есть ли функция pandas, которая сделала бы это для меня.

Для воспроизводимого примера:

toy_data #json
'{"Light":{"321":"no_light","476":"night_light","342":"lamp","454":"lamp","25":"night_light","53":"night_light","120":"night_light","346":"night_light","360":"lamp","55":"no_light","391":"night_light","243":"no_light","101":"night_light","377":"night_light","124":"no_light","368":"lamp","400":"no_light","247":"night_light","270":"lamp","208":"night_light"},"Nearsightedness":{"321":"No","476":"Yes","342":"Yes","454":"Yes","25":"No","53":"Yes","120":"Yes","346":"No","360":"No","55":"Yes","391":"Yes","243":"No","101":"No","377":"Yes","124":"No","368":"No","400":"No","247":"No","270":"Yes","208":"No"}}'

toy_data.head()
    Light       Nearsightedness
321 no_light       No
476 night_light    Yes
342 lamp           Yes
454 lamp           Yes
25  night_light    No

df = pd.DataFrame(toy_data.groupby(['Light', 'Nearsightedness']).size())

df = df.unstack('Nearsightedness')

df.columns = df.columns.droplevel()

df
Nearsightedness No  Yes
Light       
lamp             2  3
night_light      5  5
no_light         4  1

Ответы [ 2 ]

0 голосов
/ 06 октября 2018

pd.crosstab добьется цели:

pd.crosstab(df.Light, df.Nearsightedness)

Вывод:

Nearsightedness  No  Yes
Light
lamp              2    3
night_light       5    5
no_light          4    1
0 голосов
/ 06 октября 2018

Вы можете использовать pd.crosstab:

res = pd.crosstab(df['Light'], df['Nearsightedness'].eq('Yes'))

print(res)

Nearsightedness  False  True 
Light                        
lamp                 2      3
night_light          5      5
no_light             4      1
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...