Untitled

mail@pastecode.io avatar
unknown
plain_text
a year ago
1.5 kB
5
Indexable
Never
def plot_and_corr(x):
    data_temp_1 = data_actual_without_userscore.query('platform == "x" and critic_score.notna()')
    
    plt.plot(data_temp_1['critic_score'], data_temp_1['total_sales'], 'o')
    plt.xlabel('Оценка критиков')
    plt.ylabel('Суммарные продажи')
    plt.title(f'Диаграмма рассеяния: {x}')
    plt.grid(True)
    plt.show()
    corr = data_temp_1['critic_score'].corr(data_temp_1['total_sales'])
    print(f'Коэффициент корреляции мнения критиков и продаж: {corr:.4f}')
    
    data_temp_2 = data_actual_without_userscore.query('platform == "x"')
    plt.plot(data_temp_2['user_score'], data_temp_2['total_sales'], 'o')
    plt.xlabel('Оценка пользователей')
    plt.ylabel('Суммарные продажи')
    plt.title(f'Диаграмма рассеяния: {x}')
    plt.grid(True)
    plt.show()
    corr = data_temp_2['critic_score'].corr(data_temp_2['total_sales'])
    print(f'Коэффициент корреляции мнения пользователей и продаж: {corr:.4f}')

# Проходимся цклом по всем платформам
platforms = data_actual_without_userscore.query('platform != "PS4"')['platform'].unique()
for platform in platforms:
    try:
        plot_and_corr(platform)
    except TypeError:
        print(f"Ошибка: нет числовых данных для платформы {platform}")