Расчет средней зарплаты из CSV-файла — распространенная задача при анализе и обработке данных. В этой статье мы рассмотрим пять различных методов достижения этой цели с помощью Python. Каждый метод будет сопровождаться примером кода, иллюстрирующим его реализацию. Независимо от того, являетесь ли вы новичком или опытным программистом Python, в этой статье вы найдете различные подходы к расчету средней зарплаты из CSV-файла.

Метод 1: использование модуля CSV

import csv
def calculate_average_salary(csv_file):
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        salaries = [float(row['salary']) for row in reader]
        average_salary = sum(salaries) / len(salaries)
    return average_salary
# Usage example
average = calculate_average_salary('data.csv')
print("Average Salary:", average)

Метод 2: использование библиотеки Pandas

import pandas as pd
def calculate_average_salary(csv_file):
    data = pd.read_csv(csv_file)
    average_salary = data['salary'].mean()
    return average_salary
# Usage example
average = calculate_average_salary('data.csv')
print("Average Salary:", average)

Метод 3: использование библиотеки NumPy

import numpy as np
def calculate_average_salary(csv_file):
    data = np.genfromtxt(csv_file, delimiter=',', skip_header=1)
    average_salary = np.mean(data[:, 1])
    return average_salary
# Usage example
average = calculate_average_salary('data.csv')
print("Average Salary:", average)

Метод 4. Использование программы чтения CSV

import csv
def calculate_average_salary(csv_file):
    salaries = []
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        next(reader)  # Skip header
        for row in reader:
            salaries.append(float(row[1]))
    average_salary = sum(salaries) / len(salaries)
    return average_salary
# Usage example
average = calculate_average_salary('data.csv')
print("Average Salary:", average)

Метод 5: использование понимания списка

import csv
def calculate_average_salary(csv_file):
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        salaries = [float(row['salary']) for row in reader]
    average_salary = sum(salaries) / len(salaries)
    return average_salary
# Usage example
average = calculate_average_salary('data.csv')
print("Average Salary:", average)

В этой статье мы рассмотрели пять различных методов расчета средней зарплаты из CSV-файла с использованием Python. Мы обсудили различные подходы, включая встроенный модуль CSV, библиотеку Pandas, библиотеку NumPy, программу чтения CSV и понимание списков. Каждый метод имеет свои преимущества и может быть выбран исходя из конкретных требований и предпочтений. Понимая эти методы, вы сможете эффективно обрабатывать файлы CSV и извлекать полезную информацию из содержащихся в них данных.