Vitaliy_Novichikhin

8.5.1 Верно!!! 13.02.22

Feb 13th, 2022 (edited)
701
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.85 KB | None | 0 0
  1. import pandas
  2. import seaborn
  3.  
  4. data = pandas.read_csv('support_data.csv')
  5.  
  6. # названия сегментов и интервалов
  7. segments_old = ['Segment 0', 'Segment 1', 'Segment 2']
  8. segments_new = ['Потенциальные клиенты', 'Обычные клиенты', 'VIP-клиенты']
  9. intervals = ['До внедрения роботов', 'После внедрения роботов']
  10.  
  11. intervals_column = list(data['interval'])
  12. segments_column = list(data['segment']) # ваш код здесь
  13. score_column = list(data['score']) # ваш код здесь
  14. # средние оценки
  15. mean_scores = []
  16.  
  17. # ваш код здесь
  18. for index1 in segments_old:
  19.     column1 = 0
  20.     column2 = 0
  21.     column3 = 0
  22.     column4 = 0
  23.     for index2 in range(len(data)):
  24.         if segments_column[index2] == index1: #сравниваем с переменной внешнего цикла
  25.             if intervals_column[index2] == intervals[0]: #Начинанаем со значения "До"
  26.                 column1 += score_column[index2] #Вычислим сумму значений "До"
  27.                 column2 += 1 #Количество значений "До"
  28.  
  29.             else:#Продолжаем значением "После"
  30.                 column3 += score_column[index2]#Вычислим сумму значений "После"
  31.                 column4 += 1#Количество значений "После"
  32.     segment_scores = [column1 / column2, column3 / column4] #Каждое среднее значение оформим в виде списка
  33.     mean_scores.append(segment_scores) #вложим 3 полученных выше Двойных списка в общий список
  34. print(len(mean_scores))
  35. seaborn.heatmap(mean_scores, xticklabels=intervals, yticklabels=segments_new, annot=True, cmap='RdYlGn')
Add Comment
Please, Sign In to add comment