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'])
- promo_column = list(data['promo']) # ваш код здесь
- mean_promo = []
- # ваш код здесь
- for segment in segments_old: # пройдёмля по всем трём сегментам, создадим ось У
- interval_promo = [] # создадим вспомогоательный список, значения которого внесем в значения общего списка mean_prono
- for interval in intervals: # создадим ось Х для хитмэпа, пройдёмся по значениям До и После
- score = 0 # объявим переменные для осей Х и У, это сумма
- counter = 0 # количество выданных промо кодов
- for index in range(len(data)): # проидёмся по всем строкам списка
- if (segments_column[index] == segment and
- intervals_column[index] == interval):
- score += promo_column[index]
- counter += 1
- print(score, counter)
- interval_promo.append(score / counter)
- print(interval_promo)
- mean_promo.append(interval_promo)
- seaborn.heatmap(mean_promo, xticklabels=intervals, yticklabels=segments_new, annot=True, cmap='RdYlGn')
Add Comment
Please, Sign In to add comment