Untitled
unknown
plain_text
3 years ago
1.5 kB
12
Indexable
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}")Editor is loading...