Как рассчитать для каждого типа метки в столбце и сохранить их в переменной? - PullRequest
0 голосов
/ 09 ноября 2019

Мне нужно сосчитать, сколько раз 1 и 0 появляются в столбце фрейма данных. Я пробовал следующий код:

count_own_party = len(df['Deviation from Partisanship']== 1)
count_opposing_party = len(df['Deviation from Partisanship'] == 0)

print(count_own_party)
print(count_opposing_party)

Это дает одинаковый результат: 7854 для обоих значений. Хотя из graph ясно видно, что число единиц больше числа 0.

1 Ответ

0 голосов
/ 09 ноября 2019

для этого есть функция в пандах: Series.value_counts:

value_count=df['Deviation from Partisanship'].value_counts()
count_0=value_count[0]
#count_0=value_count['0'] # if it is str
count_1=value_count[1]
#count_1=value_count['1'] # if it is str

Вот пример:

print(df)
    Mnth  Income
0    Jan      80
1    Feb      80
2    Mar      50
3  April      60
4    May      60

value_count=df['Income'].value_counts()
print(value_count)



60    2
80    2
50    1
Name: Income, dtype: int64

count_60 = value_count[60]
print(count_60)
#2
...