import pandas as pd

# Загрузка файла
df = pd.read_csv("diamonds.csv")

print("Ответы теста EDA HW pract")
print("---")
print()

num_rows = df.shape[0]
print("Количество строк в датасете:", num_rows)
print("---")

missing_cut = df['Cut'].isna().sum()
print("Пропущенных значений в столбце Cut:", missing_cut)
print("---")

mean_price = round(df['Price'].mean())
print("Среднее значение Price:", mean_price)
print("---")

median_price = df['Price'].median()
print("Медианное значение Price:", median_price)
print("---")

min_carat = df['Carat Weight'].min()
max_carat = df['Carat Weight'].max()

print("минимальное и максимальное значение Carat Weight:")
print(f"{min_carat} ; {max_carat}")
print("---")

# Средняя цена по каждой форме
mean_price_by_shape = df.groupby('Shape')['Price'].mean()

# Форма с максимальной средней ценой
shape_max_price = mean_price_by_shape.idxmax()
print("Форма с самой высокой средней ценой:", shape_max_price)
print("---")

# Средняя цена по каждому цвету
mean_price_by_color = df.groupby('Color')['Price'].mean()

# Цвет с минимальной средней ценой
color_min_price = mean_price_by_color.idxmin()
print("Цвет с наименьшей средней ценой:", color_min_price)
print("---")

corr = df['Carat Weight'].corr(df['Price'])
print("Корреляция:", round(corr, 2))
print("---")

mean_depth = round(df['Depth %'].mean(), 1)
print("Среднее значение Depth %:", mean_depth)
print("---")

mean_price_by_type = df.groupby('Type')['Price'].mean()

# Тип с максимальной средней ценой
type_max_price = mean_price_by_type.idxmax()
print("Тип с более высокой средней ценой:", type_max_price)
print("---")