Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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']) # ваш код здесь
- # средние оценки
- mean_scores = []
- # ваш код здесь
- for index1 in segments_old:
- column1 = 0
- column2 = 0
- column3 = 0
- column4 = 0
- for index2 in range(len(data)):
- if segments_column[index2] == index1: #сравниваем с переменной внешнего цикла
- if intervals[index2] == 'До внедрения роботов':
- column1 += score_column[index2]
- column2 += 1
- # print(column1, column2, column3, column4)
- else:
- column3 += score_column[index2]
- column4 += 1
- # print(column1, column2, column3, column4)
- #Вывожу на экран для себя промежуточный результат, чтобы отследить ход решения задачи
- print(column1, column2, column3, column4)
- #оставляю в комментарии нижние строки до тех пор как получу значения под цикилами
- #segment_scores = [column1 / column2, column3 / column4]
- #mean_scores.append(segment_scores)
- #seaborn.heatmap(mean_scores, xticklabels=intervals, yticklabels=segments_new, annot=True, cmap='RdYlGn')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement