8.5.1 неудачная проба из прошлой задачи segments 13.02.22

import pandas
import seaborn

data = pandas.read_csv('support_data.csv')

# названия сегментов и интервалов
segments_old = ['Segment 0', 'Segment 1', 'Segment 2']
segments_new = ['Потенциальные клиенты', 'Обычные клиенты', 'VIP-клиенты']
intervals = ['До внедрения роботов', 'После внедрения роботов']

intervals_column = list(data['interval'])
segments_column = list(data['segment'])# ваш код здесь
score_column = list(data['score'])# ваш код здесь
#print(sum(score_column[:300]))
# средние оценки
mean_scores = []

# ваш код здесь
# Создаём по циклу для каждой ячейки хитмэпа До и После

for sgmnt012 in segments_column: #пройдёмся циклом по каждому значению цикла
    segment_0 = 0 #обнуляем сперва все значения для счетчика
    segment_0_sum = 0
    segment_1 = 0
    segment_1_sum = 0
    segment_2 = 0
    segment_2_sum = 0
    segment_0_list = [] #создаем списки для подсчета каждого из сегментов
    segment_1_list = []
    segment_2_list = []
    if sgmnt012 == 'Segment 0': # отсеим значение "До"
        segment_0_list.append(sgmnt012)
        segment_0 = len(segment_0_list)
    if sgmnt012 == 'Segment 1':# та же переменная, т.к. Питон читает последовательно сверху
        segment_1_list.append(sgmnt012)
        segment_1 = len(segment_1_list)
#        segment_1_sum = sum(segment_1_list[(segment_0+1):(segment_0+segment_1+1)])
    if sgmnt012 == 'Segment 2':# та же переменная, т.к. Питон читает последовательно сверху
        segment_2_list.append(sgmnt012)
        segment_2 = len(segment_2_list)
#        segment_2_sum = sum(segment_2_list[(segment_0+segment_1+1):])
#print(segment_0_list)
    print(segment_0, segment_1, segment_2)
#    for scr in
#    print(counter_after, score_after)
#print(mean_scores)
#print(len(mean_scores))
#seaborn.heatmap(mean_scores, xticklabels=intervals, yticklabels=segments_new, annot=True, cmap='RdYlGn')