Выполнение математических операций над столбцами в Python: методы и примеры кода

Для выполнения математических операций над столбцами в Python вы можете использовать различные библиотеки и методы в зависимости от структуры данных, с которой вы работаете. Вот некоторые распространенные методы с примерами кода:

  1. NumPy:
    NumPy — мощная библиотека для числовых операций в Python. Вы можете выполнять математические операции над целыми столбцами, используя массивы NumPy.
import numpy as np
# Create a 2D NumPy array
data = np.array([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])
# Perform addition on columns
sum_of_columns = np.sum(data, axis=0)
print(sum_of_columns)  # Output: [12 15 18]
# Perform multiplication on columns
product_of_columns = np.prod(data, axis=0)
print(product_of_columns)  # Output: [28 80 162]
  1. Pandas.
    Pandas — популярная библиотека для обработки и анализа данных. Он предоставляет объект DataFrame, который позволяет легко выполнять математические операции над столбцами.
import pandas as pd
# Create a DataFrame
data = {'A': [1, 4, 7],
        'B': [2, 5, 8],
        'C': [3, 6, 9]}
df = pd.DataFrame(data)
# Perform addition on columns
sum_of_columns = df.sum()
print(sum_of_columns)  # Output: A    12, B    15, C    18
# Perform multiplication on columns
product_of_columns = df.prod()
print(product_of_columns)  # Output: A     28, B    80, C    162
  1. Встроенный Python.
    Если вы работаете с простыми списками или массивами, вы можете использовать встроенные функции Python для выполнения математических операций над столбцами.
# Create a list of lists
data = [[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]]
# Perform addition on columns
sum_of_columns = [sum(col) for col in zip(*data)]
print(sum_of_columns)  # Output: [12, 15, 18]
# Perform multiplication on columns
product_of_columns = [np.prod(col) for col in zip(*data)]
print(product_of_columns)  # Output: [28, 80, 162]