58 Học Python – Tính trung bình cộng các số chính phương trong mảng

import math




# Hàm kiểm tra số chính phương

def la_so_chinh_phuong(n):

    can = int(math.sqrt(n))

    return can * can == n




# Hàm tính trung bình cộng các số chính phương trong mảng

def tinh_tb_so_chinh_phuong(mang):

    tong = 0

    dem = 0

    for so in mang:

        if la_so_chinh_phuong(so):

            tong += so

            dem += 1

   

    return tong / dem if dem > 0 else 0




# Nhập số lượng phần tử của mảng

n = int(input("Nhập số phần tử của mảng: "))

mang = [int(input(f"Nhập phần tử thứ {i+1}: ")) for i in range(n)]




# Tính trung bình cộng

ket_qua = tinh_tb_so_chinh_phuong(mang)

print(f"Trung bình cộng các số chính phương trong mảng là: {ket_qua}")

import math




def la_so_chinh_phuong(n):

    can = int(math.sqrt(n))

    return can * can == n




# Hàm tính trung bình cộng các số chính phương trong mảng (cách tối ưu)

def tinh_tb_so_chinh_phuong_toi_uu(mang):

    so_chinh_phuong = [so for so in mang if la_so_chinh_phuong(so)]

    return sum(so_chinh_phuong) / len(so_chinh_phuong) if so_chinh_phuong else 0




# Nhập số lượng phần tử của mảng

n = int(input("Nhập số phần tử của mảng: "))

mang = [int(input(f"Nhập phần tử thứ {i+1}: ")) for i in range(n)]




# Tính trung bình cộng

ket_qua = tinh_tb_so_chinh_phuong_toi_uu(mang)

print(f"Trung bình cộng các số chính phương trong mảng là: {ket_qua}")